diff --git a/Tensor Puzzlers.ipynb b/Tensor Puzzlers.ipynb
index 8c2019b..a647bbf 100644
--- a/Tensor Puzzlers.ipynb	
+++ b/Tensor Puzzlers.ipynb	
@@ -2283,12 +2283,17 @@
     "def repeat(a: TT[\"i\"], d: TT[1]) -> TT[\"d\", \"i\"]:\n",
     "    raise NotImplementedError\n",
     "\n",
-    "test_repeat = make_test(\"repeat\", repeat, repeat_spec, constraint=constraint_set)\n",
-    "\n",
+    "test_repeat = make_test(\"repeat\", repeat, repeat_spec, constraint=constraint_set)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "id": "d353c0b6-2e19-4b07-be44-a9d25d2d2b0b",
+   "metadata": {},
+   "source": [
+    "## Puzzle 21 - bucketize\n",
     "\n",
-    "# ## Puzzle 21 - bucketize\n",
-    "#\n",
-    "# Compute [bucketize](https://pytorch.org/docs/stable/generated/torch.bucketize.html)"
+    "Compute [bucketize](https://pytorch.org/docs/stable/generated/torch.bucketize.html)"
    ]
   },
   {
@@ -2414,9 +2419,9 @@
    "cell_metadata_filter": "id,-all"
   },
   "kernelspec": {
-   "display_name": "tensor_puzzles",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
-   "name": "tensor_puzzles"
+   "name": "python3"
   },
   "language_info": {
    "codemirror_mode": {
@@ -2428,7 +2433,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.10"
+   "version": "3.9.6"
   }
  },
  "nbformat": 4,