Create classify.py