Finished with the plotting of the optimization function