Setup heat map to color the wave data