working on the last three test cases