Finished docstrings, reorganized main.py