diff --git a/test.ipynb b/test.ipynb
index 25e7689..d074cb6 100644
--- a/test.ipynb
+++ b/test.ipynb
@@ -2,7 +2,7 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "id": "448d884d",
    "metadata": {},
    "outputs": [
@@ -33,7 +33,8 @@
     "print(Q.device)\n",
     "\n",
     "Q.to(\"cuda\")\n",
-    "print(Q.device)\n"
+    "print(Q.device)\n",
+    "\n"
    ]
   }
  ],