Update assignment2.s