updated comments section in main.py