HW2 complete through part 4