Added some colors in the visualization