Do sorting on the driving

Results will be relatively small (and by design must be small enough to
fit on the driver), and in large-scale testing sorting this way is
faster than first calling sortByKey