Minor tweaks to the docstrings