Added linear regression class