Docstrings and minor documentation update