Finished HW4 - Part 2.