modify code, previous tests pass