Regression analysis using Python