changed NBClassifier to class, defined classify function, fixing notebook