#define LED_1_PIN 13
#define LED_2_PIN 12
#define LED_3_PIN 11
#define LED_4_PIN 10
#define LED_5_PIN 9
#define LED_6_PIN 8

#define BUTTON_PIN 5

class Led {
private:
  byte pin;

public:
  Led(byte pin) {
    this->pin = pin;
    init();
  }

  void init() {
  pinMode(pin, OUTPUT);
  stateoff();
}

  void stateon() {
    digitalWrite(pin, HIGH);
  }

  void stateoff() {
    digitalWrite(pin, LOW);
  }
};

class Button {
private:
  byte pin;
  byte state;
  byte lastReading;
  unsigned long lastDebounceTime = 0;
  unsigned long debounceDelay = 50;

public:
  Button(byte pin) : pin(pin), state(LOW), lastReading(LOW) {
    init();
  }

  void init() {
    pinMode(pin, INPUT);
    update();
  }

  void update() {
    byte newReading = digitalRead(pin);

    if (newReading != lastReading) {
      lastDebounceTime = millis();
    }

    if (millis() - lastDebounceTime > debounceDelay) {
      state = newReading;
    }

    lastReading = newReading;
  }

  byte getState() {
    update();
    return state;
  }

  bool isPressed() {
    return (getState() == HIGH);
  }
};

class controlled {
private:
  Led led1;
  Led led2;
  Led led3;
  Led led4;
  Led led5;
  Led led6; // New LED
  Button& button;

public:
  controlled(Led& l1, Led& l2, Led& l3, Led& l4, Led& l5, Led& l6, Button& btn) :
      led1(l1), led2(l2), led3(l3), led4(l4), led5(l5), led6(l6), button(btn) {
  }

  void stateup() {
    while (button.isPressed()) {
      // Turn on LEDs in a sequence of three at a time
      led1.stateon();
      led2.stateon();
      led3.stateon();
      delay(1000);
      led1.stateoff();
      led2.stateoff();
      led3.stateoff();
      delay(1000);

      led2.stateon();
      led3.stateon();
      led4.stateon();
      delay(1000);
      led2.stateoff();
      led3.stateoff();
      led4.stateoff();
      delay(1000);

      led3.stateon();
      led4.stateon();
      led5.stateon();
      delay(1000);
      led3.stateoff();
      led4.stateoff();
      led5.stateoff();
      delay(1000);

      led4.stateon();
      led5.stateon();
      led6.stateon();
      delay(1000);
      led4.stateoff();
      led5.stateoff();
      led6.stateoff();
      delay(1000);
    }
  }

  void statedown() {
    led1.stateoff();
    led2.stateoff();
    led3.stateoff();
    led4.stateoff();
    led5.stateoff();
    led6.stateoff();
  }
};

Led led1(LED_1_PIN);
Led led2(LED_2_PIN);
Led led3(LED_3_PIN);
Led led4(LED_4_PIN);
Led led5(LED_5_PIN);
Led led6(LED_6_PIN); // New LED
Button button1(BUTTON_PIN);
controlled ledController(led1, led2, led3, led4, led5, led6, button1);

void setup() {}

void loop() {
  if (button1.isPressed()) {
    ledController.stateup();
  } else  {
    ledController.statedown();
  } 
}