#savefig('heightmap') #creates a file in the directory you ran the script from called heightmap.png