updates to torus model: making environment compatible with paper runs