Add Python documentation comments

[ifndefJOSH]

In Python, documentation comments for functions look as follows:

def foo():
    '''
A description of the function body
    '''
    # function body
    pass

Please review the functions in main.py and add comments as to what they do.