Fix docstring formatting.
Update documentation