Added main.py comments