Fix docstrings

Fix docstrings