Added some test cases in main.cpp