passes basic test cases again