Added two more test cases