fit simple linear regression in python