Added plotting of the decision boundary