Implement an ARIMA model using statsmodels (Python)