add test cases in main