Perceptron plots in python