from machine import Pin
from time import sleep

led1 = Pin(0, Pin.OUT)
led2 = Pin(1, Pin.OUT)
led3 = Pin(2, Pin.OUT)
led4 = Pin(3, Pin.OUT)
led5 = Pin(4, Pin.OUT)
x = int(input("What speed do you want(1-3)"))

if x==1:
  while True:
        led1.on()
        sleep(2)
        led1.off()
        led2.on()
        sleep(2)
        led2.off()
        led3.on()
        sleep(2)
        led3.off()
        led4.on()
        sleep(2)
        led4.off()
        led5.on()
        sleep(2)
        led5.off()
        led4.on()
        sleep(2)
        led4.off()
        led3.on()
        sleep(2)
        led1.off()
        led3.on()
        sleep(2)
        led2.off()
        led1.on()

elif x==2:
    while True:
        led1.on()
        sleep(1)
        led1.off()
        led2.on()
        sleep(1)
        led2.off()
        led3.on()
        sleep(1)
        led3.off()
        led4.on()
        sleep(1)
        led4.off()
        led5.on()
        sleep(1)
        led5.off()
        led4.on()
        sleep(1)
        led4.off()
        led3.on()
        sleep(1)
        led3.off()
        led2.on()
        sleep(1)
        led2.off()
        led1.on()

elif x==3:
    while True:
        led1.on()
        sleep(0.25)
        led1.off()
        led2.on()
        sleep(0.25)
        led2.off()
        led3.on()
        sleep(0.25)
        led3.off()
        led4.on()
        sleep(0.25)
        led4.off()
        led5.on()
        sleep(0.25)
        led5.off()
        led4.on()
        sleep(0.25)
        led4.off()
        led3.on()
        sleep(0.25)
        led3.off()
        led2.on()
        sleep(0.25)
        led2.off()
        led1.on()