Update main.py

Removed extra comments