diff --git a/06_decision_trees.ipynb b/06_decision_trees.ipynb
index 79b0f66..8e96ac4 100644
--- a/06_decision_trees.ipynb
+++ b/06_decision_trees.ipynb
@@ -86,11 +86,13 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import matplotlib as mpl\n",
+    "import matplotlib.pyplot as plt\n",
     "\n",
-    "mpl.rc('font', size=12)\n",
-    "mpl.rc('axes', labelsize=14, titlesize=14)\n",
-    "mpl.rc('legend', fontsize=14)"
+    "plt.rc('font', size=12)\n",
+    "plt.rc('axes', labelsize=14, titlesize=14)\n",
+    "plt.rc('legend', fontsize=14)\n",
+    "plt.rc('xtick',labelsize=10)\n",
+    "plt.rc('ytick',labelsize=10)"
    ]
   },
   {