Changed some tests in test_classification.py