Minor update to the docstring