Finished part 1 of HW4