diff --git a/04_training_linear_models.ipynb b/04_training_linear_models.ipynb
index 68b9f99..e6bf1d9 100644
--- a/04_training_linear_models.ipynb
+++ b/04_training_linear_models.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)"
    ]
   },
   {