Add new test cases to increse the coverage