major rework to do testing on the classification model