diff --git a/Random Forests.ipynb b/Random Forests.ipynb
index c4a7564..66732d8 100644
--- a/Random Forests.ipynb	
+++ b/Random Forests.ipynb	
@@ -258,20 +258,7 @@
     "        X.reset_index(drop=True, inplace=True)\n",
     "        y.reset_index(drop=True, inplace=True)\n",
     "        \n",
-    "        all_tree = []\n",
-    "\n",
-    "        for i in range(self.n_estimators):\n",
-    "            X_boot, y_boot = self.__bootstrap__(X, y)\n",
-    "\n",
-    "            X_boot.reset_index(drop=True, inplace=True)\n",
-    "            y_boot.reset_index(drop=True, inplace=True)\n",
-    "\n",
-    "            clf = DecisionTreeClassif(max_features=self.max_features)\n",
-    "            clf.fit(X_boot, y_boot, target_name=target_name)\n",
-    "\n",
-    "            all_tree.append(clf)\n",
-    "\n",
-    "        self.__all_tree__ = all_tree\n",
+    "        self.__all_tree__ = generate_forest(X, y, self.n_estimator)\n",
     "    \n",
     "    def __bootstrap__(self, X, y):\n",
     "        indices = np.random.randint(len(y), size=len(y))\n",