How to write if else in python