minor changes to docstring