diff --git a/mnist.ipynb b/mnist.ipynb
index 70f11da..51f13c5 100644
--- a/mnist.ipynb
+++ b/mnist.ipynb
@@ -2,13 +2,15 @@
  "cells": [
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 10,
    "metadata": {},
    "outputs": [],
    "source": [
     "import torch\n",
     "from torch import nn\n",
-    "import torch.nn.functional as F"
+    "import torch.nn.functional as F\n",
+    "from datasets import load_dataset\n",
+    "import fastcore.all as fc"
    ]
   },
   {
@@ -21,13 +23,21 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "python_main",
+   "display_name": "Python 3 (ipykernel)",
    "language": "python",
-   "name": "python_main"
+   "name": "python3"
   },
   "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
    "name": "python",
-   "version": "3.11.0"
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.8"
   },
   "orig_nbformat": 4
  },