Fix docstrings