Update some comments to match the code