Modified heatmap slightly