Changed the way user input is chosen

I changed the way user input is judged in the If statement to make the code sleeker and use less processing.