fix docs and docstrings