import time
time.sleep(0.1) # Wait for USB to become ready

print("Hello, Pi Pico!")


from gpio import *
from time import *

def main():
    pinMode(0,OUT)
    pinMode(1,IN)
    pinMode(2,OUT)

    while True:
        if digitalRead(1) == HIGH:
            customWrite(0, '2')
            customWrite(2, '1')
            digitalWrite(5, HIGH)
        else:
            customWrite(0, '0')
            customWrite(2, '2')
            digitalWrite(5, LOW)