linear regression using python code