Heatmap vis setup continued