Partitions are now a prime number to avoid imbalances