diff --git a/Trees/LinearRegression_LossFunction_Answers.ipynb b/Trees/LinearRegression_LossFunction_Answers.ipynb
index 32ab85d..a32ce3a 100644
--- a/Trees/LinearRegression_LossFunction_Answers.ipynb
+++ b/Trees/LinearRegression_LossFunction_Answers.ipynb
@@ -4,7 +4,7 @@
   "metadata": {
     "colab": {
       "provenance": [],
-      "authorship_tag": "ABX9TyMmE30JSBy91pxG0DSaosF6",
+      "authorship_tag": "ABX9TyM2yB29boJt3bLHqTG5AUqR",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -56,7 +56,7 @@
     },
     {
       "cell_type": "code",
-      "execution_count": 1,
+      "execution_count": null,
       "metadata": {
         "id": "bbF6SE_F0tU8"
       },
@@ -80,7 +80,7 @@
       "metadata": {
         "id": "UQ1qYLOeYf61"
       },
-      "execution_count": 2,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -105,7 +105,7 @@
       "metadata": {
         "id": "785wp16FYjt5"
       },
-      "execution_count": 5,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -141,7 +141,7 @@
       "metadata": {
         "id": "aD2g9L9hd5Bd"
       },
-      "execution_count": 6,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -157,7 +157,7 @@
           "height": 458
         }
       },
-      "execution_count": 8,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "display_data",
@@ -188,7 +188,7 @@
       "cell_type": "code",
       "source": [
         "# Function to calculate the loss\n",
-        "def compute_loss(x,y,phi0,phi1):\n",
+        "def compute_loss(x,y,f,phi0,phi1):\n",
         "\n",
         "  # TODO Replace this line with the loss calculation (equation 2.5)\n",
         "  loss = 0\n",
@@ -203,7 +203,7 @@
       "metadata": {
         "id": "PEenxRo3ePCL"
       },
-      "execution_count": 11,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -220,7 +220,7 @@
           "base_uri": "https://localhost:8080/"
         }
       },
-      "execution_count": 13,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "stream",
@@ -275,7 +275,7 @@
       "metadata": {
         "id": "zlUXJ-5zhlb9"
       },
-      "execution_count": 16,
+      "execution_count": null,
       "outputs": []
     },
     {
@@ -291,7 +291,7 @@
           "height": 551
         }
       },
-      "execution_count": 17,
+      "execution_count": null,
       "outputs": [
         {
           "output_type": "display_data",