Added code for SVM classification