Create 4. while for handling user input.py