Classifier.py: Fixed Bug