Working on linear regression. Saving work in progress