Update as per HW4