update test/main.cpp