Using set insted of map