remove test code in main