updated exercise 2 test cases