diff --git a/CRF.ipynb b/CRF.ipynb
index 2516854..dea2b3c 100644
--- a/CRF.ipynb
+++ b/CRF.ipynb
@@ -7,7 +7,7 @@
     "# Task B: Named Entity Recognition with CRF on Hindi Dataset. (Total: 60 Points out of 100)\n",
     "\n",
     "In this part, you will use a CRF to implement a named entity recognition tagger.\n",
-    "We have implemented a CRF for you in crf.py along with some functions to build, and pad feature vectors. Your job is to add more features to learn a better tagger. Then you need to complete the traiing loop implementation.\n",
+    "We have implemented a CRF for you in crf.py along with some functions to build, and pad feature vectors. Your job is to add more features to learn a better tagger. Then you need to complete the training loop implementation.\n",
     "\n",
     "Finally, you can checkout the code in `crf.py` -- reflect on CRFs and span tagging, and answer the discussion questions.\n",
     "\n",
@@ -268,7 +268,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from crf import f1_score, predict, PAD_SYMBOL\n",
+    "from crf import f1_score, predict, pad_features, pad_labels, PAD_SYMBOL\n",
     "\n",
     "# TODO: Implement the training loop\n",
     "# HINT: Build upon what we gave you for HW2.\n",