touched up docstrings