Document Code

We need someone to document the code. I would suggest using [this helpful guide](https://realpython.com/documenting-python-code/) to do so. Please add comments where necessary and docstrings for every class function.