Changes to Part 3

Created a unit test function