Added option for multiple heatmaps (UI is still a bit rough)