Update classifier.py