python user input and if statement