Added awesome comments to main.py