Create classification.py