Updating docs and docstrings