Update assignment4.s