Added comments for main.py