Проблема кодирования Arduino: 2 секции не работают при объединении

Я использую Arduino Uno и пытаюсь объединить две части школьного проекта. Обе части работают нормально по отдельности, но в сочетании работает только часть 1. Часть 1: Включает в себя 1 кнопочный переключатель, 2 светодиода, один из которых горит, а другой не горит, кнопка переключается между тем, что включено или выключено. Часть 2: Сервопривод, управляемый потенциометром

Я определил, что сервопривод работает только при нажатии кнопки, чего не должно происходить, код потенциометра/сервопривода не должен зависеть от кода для светодиодов. Я определенно новичок в этом и не понимаю многого из того, что делает код, поэтому простите меня за избыточность. Буду признателен за любую оказанную помощь.

#include <Servo.h>
Servo myServo;
int const potPin = A0;
int potVal;
int angle;
int switchState;
int lastSwitchState = 0;
const int bluePin = 3;
const int yellowPin = 4;
const int button = 2;
int bluelight = LOW;
int yellowlight = HIGH;

void setup(){
  pinMode(bluePin, OUTPUT); //blue LED
  pinMode(yellowPin, OUTPUT); //yellow LED
  pinMode(button, INPUT); //switch
  myServo.attach(9);
  Serial.begin(9600);
}
void loop(){
  // PART 1 - this is the only part that seems to be working now?
  switchState = digitalRead(button);
  while (digitalRead(button)==LOW); 
  if (digitalRead(button)==LOW){ 
  bluelight=!bluelight;
    digitalWrite(bluePin, bluelight);
    digitalWrite(yellowPin, yellowlight);
  }
  else{
  if (switchState=!lastSwitchState) {
    yellowlight=!yellowlight;
    bluelight=!bluelight;
  digitalWrite(yellowPin, yellowlight);
  digitalWrite(bluePin, bluelight);
  }
  }
  // PART 2 - Only works when button is pressed? 
    potVal = analogRead(potPin);
  Serial.print("potVal: ");
  Serial.print(potVal);
  angle = map(potVal, 0, 1023, 0, 179);
  Serial.print(", angle: ");
  Serial.println(angle);
  myServo.write(angle);

  delay(250); //wait for a quarter second
  }