Better coloring + Legend mockup