Update main.py

changed comments