add test cases in main function