Updated driver to use Terrain