added somewhat working heatmap