Update classification.cpp