Updated linear regression

Added ML course wrapper functions, removed extraneous variable
declarations and added code to display the plot