add test for the problem 3