Update classifier_main.cpp