Add repartition method to Data