Create if_else_while.py