Added class to test SVM