Added 2 more test cases