Update #04

+ Added while loop to main.py