Update comments to match code