Update main.py

add some comments