Update main.py

removed comments