Add more tests to main.cpp