changes to classifier.py