fixed plotting and decision boundary stuff