Update main.py

Removed some comments and added one.