Todo

Add docs and docstrings