Added comments.
Small code cleanup.