working on the classifier