Added extra test cases