Trying to get some kind of plot for the decision boundary