DOC better docstring