Update code for finalize update