Update fluff_heatmap.py