cleanup and confusion matrix
documentation