editing classify.py