linear regression python code with dataset