Update app.py

app.py

@@ -15,9 +15,7 @@ def get_userInput():
 
 def validate_userInput(user_input):
     if user_input.isdigit():
-        if int(user_input) < 1 or int(user_input) > 9:                
-            return False
-        else: 
+        if int(user_input) >= 1 and int(user_input) <= 9:                
             if int(user_input) in st.session_state.legal_numbers:
                 return True
             else:
@@ -125,13 +123,14 @@ st.header("Python - Tic Tac Toe")
 display_guide()
 
 user_input=get_userInput()
-update_legalNumbers(0)
-#display_board(0)
 
 #UI Buttons
 submit=st.button('Submit')
 replay=st.button('Replay')
 
+update_legalNumbers(0)
+display_board(0)
+
 #Button functionality
 if submit:
     response=validate_userInput(user_input)