updated tests in main.cpp