add tests for Problem 3