Heat map - correcting axes