Update btns.py

btns.py

 import RPi.GPIO as GPIO # Import Raspberry Pi GPIO library
 
-def main():
-  # Main program block
-  GPIO.setwarnings(False) # Ignore warning for now
-  GPIO.setmode(GPIO.BOARD) # Use physical pin numbering
-  GPIO.setup(13, GPIO.IN, pull_up_down=GPIO.PUD_DOWN) # Set pin 22 to be an input pin and set initial value to be pulled low (off)
-  GPIO.setup(15, GPIO.IN, pull_up_down=GPIO.PUD_DOWN) # Set pin 23 to be an input pin and set initial value to be pulled low (off)
-  GPIO.setup(16, GPIO.IN, pull_up_down=GPIO.PUD_DOWN) # Set pin 24 to be an input pin and set initial value to be pulled low (off)
-  GPIO.setup(18, GPIO.IN, pull_up_down=GPIO.PUD_DOWN) # Set pin 26 to be an input pin and set initial value to be pulled low (off)
-  while True: # Run forever
-      if GPIO.input(13) == GPIO.HIGH:
-          print("no")
-          break
-      if GPIO.input(15) == GPIO.HIGH:
-          print("down")
-          break
-      if GPIO.input(16) == GPIO.HIGH:
-          print("up")
-          break
-      if GPIO.input(18) == GPIO.HIGH:
-          print("yes")
-          break
-  
-  return
-#message = input("Press enter to quit\n\n") # Run until someone presses enter
-
-
-
-if __name__ == '__main__':
 
   try:
-    main()
+    GPIO.setwarnings(False) # Ignore warning for now
+    GPIO.setmode(GPIO.BOARD) # Use physical pin numbering
+    GPIO.setup(13, GPIO.IN, pull_up_down=GPIO.PUD_DOWN) # Set pin 22 to be an input pin and set initial value to be pulled low (off)
+    GPIO.setup(15, GPIO.IN, pull_up_down=GPIO.PUD_DOWN) # Set pin 23 to be an input pin and set initial value to be pulled low (off)
+    GPIO.setup(16, GPIO.IN, pull_up_down=GPIO.PUD_DOWN) # Set pin 24 to be an input pin and set initial value to be pulled low (off)
+    GPIO.setup(18, GPIO.IN, pull_up_down=GPIO.PUD_DOWN) # Set pin 26 to be an input pin and set initial value to be pulled low (off)
+    while True: # Run forever
+        if GPIO.input(13) == GPIO.HIGH:
+            print("no")
+            break
+        if GPIO.input(15) == GPIO.HIGH:
+            print("down")
+            break
+        if GPIO.input(16) == GPIO.HIGH:
+            print("up")
+            break
+        if GPIO.input(18) == GPIO.HIGH:
+            print("yes")
+            break
   finally:
     GPIO.cleanup()