diff --git a/baseline-retrieval-system/bigramsrefactor.ipynb b/baseline-retrieval-system/bigramsrefactor.ipynb
index f8fd3fd..cfa81c9 100644
--- a/baseline-retrieval-system/bigramsrefactor.ipynb
+++ b/baseline-retrieval-system/bigramsrefactor.ipynb
@@ -75,7 +75,7 @@
     }
    ],
    "source": [
-    "!pip3 install tira ir-datasets python-terrier"
+    "!pip3 install tira ir-datasets python-terrier nltk pandas"
    ]
   },
   {
@@ -186,7 +186,7 @@
     "#Create List from our docs iterator\n",
     "#docs = list(docs)\n",
     "#cutoff the list after the first 500 documents\n",
-    "docs = docs[:500]\n",
+    "#docs = docs[:500]\n",
     "\n",
     "#Print some example documents\n",
     "for i,doc in enumerate(docs):\n",