Update main.py with prettier input request