Adding in 2 test cases