changes in classify.py