Style fixes.
Update some docstrings and comments.