Updated the test (main.cpp)