Adding forecasting capability