Added some docstrings + added main class documentation to docs