add strain axis to heatmap