Add heatmap-like coloring to draw