Update comments for ease of understanding.

See #7