Updates to GetPartition