updated comment in classifier.py