Add test in main.cpp