diff --git a/Lambda New Features.ipynb b/Lambda New Features.ipynb
index a5f8d59..96ce44d 100644
--- a/Lambda New Features.ipynb	
+++ b/Lambda New Features.ipynb	
@@ -58,17 +58,13 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 9,
+   "execution_count": 13,
    "metadata": {},
    "outputs": [],
    "source": [
     "def word_count(review, word):\n",
-    "    cont = 0\n",
     "    review = review.lower()\n",
-    "    if word in review.lower():\n",
-    "        return review.count(word)\n",
-    "    else:\n",
-    "        return 0"
+    "    return review.count(word)"
    ]
   },
   {