#include "Led.h"
#include "PushButton.h"

#define PB1 2
#define RED_LED_PIN 11
#define YELLOW_LED_PIN 10
#define GREEN_LED_PIN 9

Led redLed(RED_LED_PIN);
Led yellowLed(YELLOW_LED_PIN);
Led greenLed(GREEN_LED_PIN);
PushButton Button(PB1, 1, 1);

void setup() {
  //Serial.begin(115200);
  redLed.init();
  yellowLed.init();
  greenLed.init();
  Button.init();
}

void loop() {
  if(Button.isPressed()){
    //Serial.println("Button is pressed");
    redLed.on();
    yellowLed.off();
    greenLed.on();
  } else {
    //Serial.println("Button is not pressed");
    redLed.off();
    yellowLed.on();
    greenLed.off();
  }
  delay(100);
}