Improve networks heatmap gallery example