Treemap: colorAxis should now work properly in treemap