Added the computation of the cost.
Added the linear fit learned to the data plot.