Reverted non-essential changes
Reverted reformatting changes in arima_model
Fixed an additional issue when forecasting
Cleaned up the demo code