Final version, does confusion matrix