Permission change on example data sets
Avoid shuffle to get distinct keys, result is small enough to fit on driver