Added linear regression code