Removed decision boundary plotting