Update test_main.cpp