Make the extension dimensions more dynamic.
Now to add a new color simply add that to colorMap.