Update state_heatmap.r