finish first 2 with tests in main()
finish hw 4