DOC: better docstrings