minor changes in docstring