HW04: think I finished part 3