Demo uses new plot.weight.classes