added 2 extra test cases