Updates for HW 4