Use statsmodels to add constants