Extract classification.py