replaced the load inside classification_test to outside