void setup() {
  pinMode(1, OUTPUT);
  pinMode(2, INPUT);
}

void loop() {
  byte button = digitalRead(2);

  if (button == HIGH) {
    digitalWrite(1, HIGH);
  }
  else {
    digitalWrite(2, LOW);
  }
}