added heatmap code and sample heatmaps