Add test cases to increase code coverage