diff --git a/assignment3/Assignment3.ipynb b/assignment3/Assignment3.ipynb
index 755c595..e49f6aa 100644
--- a/assignment3/Assignment3.ipynb
+++ b/assignment3/Assignment3.ipynb
@@ -663,6 +663,7 @@
         "    \"\"\"Returns the value of rosenbrock's function and its gradient at x\n",
         "    \"\"\"\n",
         "    val = TODO\n",
+        "    # Gradient should be np.array\n",
         "    dVdX= TODO\n",
         "    return [val, dVdX]\n",
         "\n",