Using if in Python