Add svm trainning and predicting code