Modify method of plotting the decision boundary