Fix errors in plot_results