Prob_1.2, plot_legend_loc modified