improved results of plot_trace example