Update test_class.cpp