updated assignment. works for all test cases right now