diff --git a/Week4.ipynb b/Week4.ipynb
index ba6a247..a5bf67f 100644
--- a/Week4.ipynb
+++ b/Week4.ipynb
@@ -457,17 +457,7 @@
    ],
    "source": [
     "def highlight_word(sentence, word):\n",
-    "  ret = \"\"\n",
-    "  l = len(word)\n",
-    "  words = sentence.split()\n",
-    "  for i,w in enumerate(words):\n",
-    "    if w[:l] == word:\n",
-    "      ret += w[:l].upper()+w[l:]\n",
-    "    else:\n",
-    "      ret += w\n",
-    "    if i != len(words)-1:\n",
-    "      ret += \" \"\n",
-    "  return ret\n",
+    "  return sentence.replace(word,word.upper())\n",
     "\n",
     "print(highlight_word(\"Have a nice day\", \"nice\"))\n",
     "print(highlight_word(\"Shhh, don't be so loud!\", \"loud\"))\n",