Linear Regression code