Added two test cases