Add test for problem 1