Python programing a User Input with an IF Statement