Initial commit for part 2 of the assignment