Create more robust heatmap example