#include <Wire.h>
#include <Adafruit_GFX.h>
#include <Adafruit_SSD1306.h>

#define SCREEN_WIDTH 128
#define SCREEN_HEIGHT 64

Adafruit_SSD1306 display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire);

void setup() {
  pinMode(1, OUTPUT);
  display.begin(SSD1306_SWITCHCAPVCC, 0x3C);
  display.clearDisplay();
  display.setTextSize(1);
  display.setTextColor(SSD1306_WHITE);
  display.setCursor(0,0);
  display.println(F("Traffic Light"));
  display.display();
}
int time = 10;
String stat = "Stop";
void loop() {
  display.clearDisplay();
  display.setCursor(0, 0);
  display.println(stat + ": " + time);
  display.display();
  if (time > 1) {
    digitalWrite(2, LOW);
    if (stat == "Stop") {
      digitalWrite(1, HIGH);
      digitalWrite(3, LOW);
    } else {
      digitalWrite(1, LOW);
      digitalWrite(3, HIGH);
    }
  } else if (time == 1) {
    digitalWrite(1, LOW);
    digitalWrite(2, HIGH);
    digitalWrite(3, LOW);
  }
  time -= 1;
  if (time == 0) {
    time = 10;
    if (stat == "Stop") {
      stat = "Walk";
    } else {
      stat = "Stop";
    }
  }
  delay(1000);
}