add_diag_names function

Add diagonal names to heatmap.