Update btns.py

btns.py

@@ -21,6 +21,8 @@ def main():
       if GPIO.input(18) == GPIO.HIGH:
           print("yes")
           break
+  
+  return
 #message = input("Press enter to quit\n\n") # Run until someone presses enter
 
 
@@ -29,6 +31,5 @@ if __name__ == '__main__':
 
   try:
     main()
-    return
   finally:
     GPIO.cleanup()