minor changes. test passes.