Added some more documentation to main.py