Updated barchart to include test set