added linear regression, it need clean-up