DOC adjust docstring as suggested by @gvaroquaux