Part 1 of HW 5 finished