Part 2 now using sets