update test cases for problem 9