Partition process stage improvements