Changed comments to reflect code changes