Update assignment-2