problem2 unit test