Worked on if statments in python