Added some stats to the code to check the performance of the classification.