Update assignment-4