diff --git a/ML_lab2_word_count_student.ipynb b/ML_lab2_word_count_student.ipynb
index 8b9d63e..001361c 100644
--- a/ML_lab2_word_count_student.ipynb
+++ b/ML_lab2_word_count_student.ipynb
@@ -608,9 +608,9 @@
     "        str: The cleaned up string.\n",
     "    \"\"\"\n",
     "    <FILL IN>\n",
-    "print removePunctuation('Hi, you!')\n",
-    "print removePunctuation(' No under_score!')\n",
-    "print removePunctuation(' *      Remove punctuation then spaces  * ')"
+    "print removePunctuation(u'Hi, you!')\n",
+    "print removePunctuation(u' No under_score!')\n",
+    "print removePunctuation(u' *      Remove punctuation then spaces  * ')"
    ]
   },
   {
@@ -622,8 +622,8 @@
    "outputs": [],
    "source": [
     "# TEST Capitalization and punctuation (4b)\n",
-    "Test.assertEquals(removePunctuation(\" The Elephant's 4 cats. \"),\n",
-    "                  'the elephants 4 cats',\n",
+    "Test.assertEquals(removePunctuation(u\" The Elephant's 4 cats. \"),\n",
+    "                  u'the elephants 4 cats',\n",
     "                  'incorrect definition for removePunctuation function')"
    ]
   },