DOC: updated docstring