How to write IF ELSE ELIF statement in Python