rewrite the forestTrain
a