modify classify.py file