Corrected some comments to match the code