Adding heatmap example