 
 void updateServo() {
   if (servo_state == 0) {
-    if (servo_pos <= 90) {
+    if (servo_pos < 90) {
       servo_pos += 1;
       servo.write(servo_pos);
     }
   }
   else if (servo_state == 1) {
-    if (servo_pos <= 90) {
-      servo_pos += 1;
+    if (servo_pos > 0) {
+      servo_pos -= 1;
       servo.write(servo_pos);
     } 
     else {

    }
  }
  else if (servo_state == 2) {
    if (servo_delay >= 200) {
      servo_state = 3;
    } 
    else {


  // Initialize LED output
  pinMode(LED_PIN, OUTPUT);

  // Initialize button input
  pinMode(BUTTON_PIN, INPUT);

  // Initialize servo
  servo.attach(SERVO_PIN);

  // Initialize speaker
  pinMode(SPEAKER_PIN, OUTPUT);
  
  beep(50);
  beep(50);
  beep(50);

  Serial.println("Setup complete");
}
