Changed set call for backwards compatibility
Add repartition method to Data