Update code for final round