Change heatmap import