generalized - add heatmap
generalized - improve heat map