Fix comments.
Minor clean-up.