Added more tests for part2.
Fixed a copy paste error.