Changed to linear svm