Updated the finish section2 code