Made updates to plot_result; Still needs some fine tuning