Update assignment4