updatig heatmap and notebook