Adding some changes to the HeightMap stuff