fixed test cases. all tests are currently passing