Add new test cases to increase coverage