Changed the HashMap to Sets