Fix the comments to reflect the function of the code.