Part 1 of assignment with some very basic testing