begin problem 3, make test first