Closes #4 - Addd docstrings for the classes and functions