Heatmap looks prettier now