plot_class_allocation function

Plot class allocation (for P_r = 2 only)