Added docstrings to classes and functions
Removed unused class attributes