Updated docstrings for sphinx