docstring cleanup
Fixing up docstrings and adding fancy sphinx docs