simplified the return
bad first draft. doesn't pass all test cases