Chages to SVM model