Slight change to docstring