Improve docstrings for Sphinx