Make use of the new get_partition_number function