add main() and test()