add test code in main