Update main.py

blah-blah