Minor changes to the plot_results.py