tests for part2 updated