write a simple if statement in python