Added docstrings to classes and methods