exercise for if else in python