HW 4 complete