Changed NB to SVM