add some test code on main()