Change main example function into a parabola and plot it
Some formatting changes