added legend possibility in plot_evolution