Prettify and update colors in graph.py