Fixed bug with decision boundary plotting