Changes added to classify.py