plot_color_dist can now accept var. arg