using if statements in python