Add DecionTree and RandomForest methods