Add comments to revised code