change back to our classifier instead of svm