int led_red = 2;
int led_yellow = 3;
int led_green = 4;
int btn_pin = 12;

boolean ledOn = HIGH;

void setup()
{
  pinMode(led_red, OUTPUT);
  pinMode(led_yellow, OUTPUT);
  pinMode(led_green, OUTPUT);
}

void loop()
{
  digitalWrite(led_yellow, LOW);
  digitalWrite(led_red, HIGH);
  for (int i = 0 ; i < 50 ; i++) {
    if (digitalRead(btn_pin)) 
      break;
    delay(40);
  }

  digitalWrite(led_red, LOW);
  digitalWrite(led_green, HIGH);
  delay(2000);          

  digitalWrite(led_green, LOW);
  digitalWrite(led_yellow, HIGH);
  for (int i = 0 ; i < 10 ; i++) {
    if (digitalRead(btn_pin)) 
      break;
    delay(70);
  }
}