Added 2 test cases