[add] base layer for color predictions
[add] more color stuff
[add] some tweaks to the visualization