Add comments to codes