added confusion matrix stuff and testing on test data