Improve documentation.

- Small cosmetic code fixes