diff --git a/reinforcement-learning/mlx-policy-based.ipynb b/reinforcement-learning/mlx-policy-based.ipynb
index 48e970a..5379acb 100644
--- a/reinforcement-learning/mlx-policy-based.ipynb
+++ b/reinforcement-learning/mlx-policy-based.ipynb
@@ -25,7 +25,6 @@
    "source": [
     "import time\n",
     "import imageio\n",
-    "import numpy as np\n",
     "import mlx.nn as nn\n",
     "import mlx.core as mx\n",
     "import gymnasium as gym\n",
@@ -232,7 +231,7 @@
     "    ax[0].axvline(x=i, color='gray', linewidth=0.5)\n",
     "    ax[1].axvline(x=i, color='gray', linewidth=0.5)\n",
     "\n",
-    "plt.savefig(\"./static/images/torch-pbg.png\")\n",
+    "plt.savefig(\"./static/images/mlx-pbg.png\")\n",
     "plt.show()"
    ],
    "metadata": {
@@ -270,7 +269,7 @@
     "    state = mx.array(state)\n",
     "\n",
     "    images.append(environment.render())\n",
-    "_ = imageio.mimsave('./static/images/torch-pbg.gif', images, duration=25)"
+    "_ = imageio.mimsave('./static/images/mlx-pbg.gif', images, duration=25)"
    ],
    "metadata": {
     "collapsed": false,
@@ -284,7 +283,7 @@
   {
    "cell_type": "markdown",
    "source": [
-    "<img src=\"./static/images/torch-pbg.gif\" width=\"1000\" height=\"1000\" />"
+    "<img src=\"./static/images/mlx-pbg.gif\" width=\"1000\" height=\"1000\" />"
    ],
    "metadata": {
     "collapsed": false
diff --git a/reinforcement-learning/mlx-value-based.ipynb b/reinforcement-learning/mlx-value-based.ipynb
index d5e4d3a..d233401 100644
--- a/reinforcement-learning/mlx-value-based.ipynb
+++ b/reinforcement-learning/mlx-value-based.ipynb
@@ -23,10 +23,8 @@
    },
    "outputs": [],
    "source": [
-    "import copy\n",
     "import time\n",
     "import imageio\n",
-    "import mlx.nn as nn\n",
     "import mlx.core as mx\n",
     "import gymnasium as gym\n",
     "import mlx.optimizers as optim\n",
@@ -296,7 +294,7 @@
     "    ax[0].axvline(x=i, color='gray', linewidth=0.5)\n",
     "    ax[1].axvline(x=i, color='gray', linewidth=0.5)\n",
     "\n",
-    "plt.savefig(\"./static/images/torch-dqn.png\")\n",
+    "plt.savefig(\"./static/images/mlx-dqn.png\")\n",
     "plt.show()"
    ],
    "metadata": {
@@ -333,7 +331,7 @@
     "    state = mx.array(state)\n",
     "\n",
     "    images.append(environment.render())\n",
-    "_ = imageio.mimsave('./static/images/torch-dqn.gif', images, duration=25)"
+    "_ = imageio.mimsave('./static/images/mlx-dqn.gif', images, duration=25)"
    ],
    "metadata": {
     "collapsed": false,
@@ -347,7 +345,7 @@
   {
    "cell_type": "markdown",
    "source": [
-    "<img src=\"./static/images/torch-dqn.gif\" width=\"1000\" height=\"1000\" />"
+    "<img src=\"./static/images/mlx-dqn.gif\" width=\"1000\" height=\"1000\" />"
    ],
    "metadata": {
     "collapsed": false