Update for HW4