Added the option to make a random forest instead of a tree