minor tweak to classify.py