Create while_loop.py