More docs and docstrings edits