from machine import Pin
import time
import ssd1306
time.sleep(0.1) # Wait for USB to become ready
red_led=Pin(8,Pin.OUT)
green_led=Pin(6,Pin.OUT)
yellow_led=Pin(7,Pin.OUT)
i2c = machine.I2C(0, sda=machine.Pin(4), scl=machine.Pin(5), freq=400000)
oled = ssd1306.SSD1306_I2C(128, 64, i2c)
while True:
    red_led.value(1)
    green_led.value(0)
    yellow_led.value(0)
    oled.fill(0)
    oled.text("RED LIGHT ON", 20, 25)
    oled.show()
    
    time.sleep(2)
    red_led.value(0)
    green_led.value(1)
    yellow_led.value(0)
    oled.fill(0)
    oled.text("GREEN LIGHT ON", 15, 25)
    oled.show()
    time.sleep(2)
    red_led.value(0)
    green_led.value(0)
    yellow_led.value(1)
    oled.fill(0)
    oled.text("YELLOW ON", 20, 25)
    oled.show()
    time.sleep(2)