minor changes to the docstring