HW4 finished v3