could  you add more test cases?