Add forecasting support