add updated heatmap notebook