added repartition api