DOC: fix up docstrings