Added example heatmaps from a small simulated model