diff --git a/assignment.ipynb b/assignment.ipynb
index 15eea39..9fcd8ca 100644
--- a/assignment.ipynb
+++ b/assignment.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 104,
+   "execution_count": 106,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -71,7 +71,7 @@
     "        (String) height\n",
     "\n",
     "    Returns:\n",
-    "        Generator[Cat]: A list of cat objects from the JSON object.\n",
+    "        Generator[Cat]: A Generator object creating cat objects from the JSON object.\n",
     "    \"\"\"\n",
     "\n",
     "    for cat in cat_json:\n",