Simpler heatmap example. Fixes #672