Update plot_heat_map.R