finished coding on heatmap