Fix color key for rotation and fix heatmap lables