diff --git a/lstm_word2vec.ipynb b/lstm_word2vec.ipynb
index d839e7e..376c64b 100644
--- a/lstm_word2vec.ipynb
+++ b/lstm_word2vec.ipynb
@@ -34,7 +34,7 @@
     "editable": true
    },
    "source": [
-    "Download the Amazon reviews training data from a public Azure blob"
+    "Download the Amazon reviews data from a public Azure blob"
    ]
   },
   {
@@ -48,7 +48,8 @@
    "outputs": [],
    "source": [
     "CONTAINER_URL = \"https://anargyridsa.blob.core.windows.net/dlvm/\"\n",
-    "trainFile = \"amazon_reviews_train.csv\""
+    "trainFile = \"amazon_reviews_train.csv\"\n",
+    "testFile = \"amazon_reviews_test.csv\""
    ]
   },
   {
@@ -61,8 +62,9 @@
    },
    "outputs": [],
    "source": [
-    "# read\n",
-    "train_data = pd.read_csv(CONTAINER_URL + trainFile, header=None, names=['rating', 'title', 'text'])"
+    "# read data\n",
+    "train_data = pd.read_csv(CONTAINER_URL + trainFile, header=None, names=['rating', 'title', 'text'])\n",
+    "test_data = pd.read_csv(CONTAINER_URL + testFile, header=None, names=['rating', 'title', 'text'])"
    ]
   },
   {
@@ -143,7 +145,9 @@
    "cell_type": "code",
    "execution_count": null,
    "metadata": {
-    "collapsed": true
+    "collapsed": true,
+    "deletable": true,
+    "editable": true
    },
    "outputs": [],
    "source": [