added comments to main.py