Add test file for part2 of problem