update dt_train_multi function header to include Y and Z