Added stage to partition choice for 2nf