simple if else program questions in python