Working with classifier, almost done