Fix docstrings for sphinx