write initial code - all tests pass