Basic Linear Classifier Python