As mentioned in #12, we want to split the data in the different partitions, and need to decide how to.