added forecasts to train