update train_SVM method