Fixes to docstring