All test cases working