Updating the code to kind of use a heightmap