add functions to draw heatplot