diff --git a/Notebooks/Chap08/8_2_Bias_Variance_Trade_Off.ipynb b/Notebooks/Chap08/8_2_Bias_Variance_Trade_Off.ipynb
index ad2177c..ff5b66b 100644
--- a/Notebooks/Chap08/8_2_Bias_Variance_Trade_Off.ipynb
+++ b/Notebooks/Chap08/8_2_Bias_Variance_Trade_Off.ipynb
@@ -4,7 +4,6 @@
   "metadata": {
     "colab": {
       "provenance": [],
-      "authorship_tag": "ABX9TyPz1B8kFc21JvGTDwqniloA",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -185,10 +184,8 @@
         "          if A[i,j] < 0:\n",
         "              A[i,j] = 0;\n",
         "\n",
-        "  ATA = np.matmul(np.transpose(A), A)\n",
-        "  ATAInv = np.linalg.inv(ATA)\n",
-        "  ATAInvAT = np.matmul(ATAInv, np.transpose(A))\n",
-        "  beta_omega = np.matmul(ATAInvAT,y)\n",
+        "  beta_omega = np.linalg.lstsq(A, y, rcond=None)[0]\n",
+        "\n",
         "  beta = beta_omega[0]\n",
         "  omega = beta_omega[1:]\n",
         "\n",