Updated comments to match code changes