Minor edits to docstring