Added toggle buttons for the heatmaps