minor fix, color in plot_dist