finish iteration 2 code