Update main.py

comment describing about working of code