linear regression example problems python