Update to terrain.py