Correct gradient scale for heatmap1.py.