Added docstrings

Added docstrings to all classes and functions.