Keep the keys of the input data set

To refer back to the original data of the main application, keep the
keys in the data set, and use those keys in the buckets returned.