Minor changes to docstring