Improvements to docstrings