Better generate_heatmap

[dorimedini]

• Add a title
• Set the scale to [0,1] (fix it so we can compare separate graphs)
• Fix a color scheme (ditto)