Correct comments to reflect functionality