Added label loading for the confusion matrix.