Update main.cpp

updated tests