Minor comments to classifier.py