Update main.py

modify such that real user input are used