hw2 except for heatmap