add test in main