Add docs setup
Add docstrings