Working for some test cases