Add comment.
Clean up code.