Lots of updates to docs and docstrings