Add x-axis to heatmap, need to move and add labels