Doc additions for train and forecast