modified classifier.py