working on new heatmap function