Added some documentation to main.py