Added code for plotting SVM