diff --git a/ProgrammingAssignment_1/ProgrammingAssignment1.ipynb b/ProgrammingAssignment_1/ProgrammingAssignment1.ipynb
index fed5455b528f278bb12f4c91b5e49e4959cb626c..62d7633202aea1bc6377448376d54f1f0b250b19 100644
--- a/ProgrammingAssignment_1/ProgrammingAssignment1.ipynb
+++ b/ProgrammingAssignment_1/ProgrammingAssignment1.ipynb
@@ -218,7 +218,7 @@
     "        '''\n",
     "        raise NotImplementedError\n",
     "        \n",
-    "        pred = []\n",
+    
     "        # TODO\n",
     "        \n",
     "        # for each point in test_features\n",
@@ -229,7 +229,7 @@
     "        # you'll need proportion of the dominant class\n",
     "        # in k nearest neighbors\n",
     "        \n",
-    "        return np.array(pred)\n",
+    "        return preds\n",
     "    "
    ]
   },