Adding more test cases