Put all test code in main()