Improvements to the docstring