Created wrapper to train all 6 trees