Added height map code