Regression using Python