Adjusting the *_classification.py file.