Update main.py

Added comments