Adding more tests in main.cpp