Update comments.
Code cleanup.