# Import required libraries
from time import sleep
from machine import Pin
import leds

# pull-up switches connected to pins 6 (r1) & 2 (r2)
sw1 = Pin(6, Pin.IN, Pin.PULL_UP)
sw2 = Pin(2, Pin.IN, Pin.PULL_UP)

leds.createleds(9, 8, 7, 5, 4, 3)

while True:
    try:
        leds.display(0b001100)
        sleep(3)
        while sw2.value():
            sleep(0.1)
        leds.display(0b010100)
        sleep(1)
        leds.display(0b100100)
        sleep(1)
        leds.display(0b100110)
        sleep(1)
        leds.display(0b100001)
        sleep(3)
        while sw1.value():
            sleep(0.1)
        leds.display(0b100010)
        sleep(1)
        leds.display(0b100100)
        sleep(1)
        leds.display(0b110100)
        sleep(1)
    except KeyboardInterrupt:
        break
leds.display(0b000000)
print("Finished.")