Changed comments to reflect code