Updates to visualize SVM