added forecast to train