Minor improvements to docs.
Fix docstrings.