Update assignment2