diff --git a/code/notebooks/stft.ipynb b/code/notebooks/stft.ipynb
index 7a86841..b018033 100644
--- a/code/notebooks/stft.ipynb
+++ b/code/notebooks/stft.ipynb
@@ -334,8 +334,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "len(ready_data1a)\n",
-    "# plt.pcolormesh(ready_data1[0])"
+    "# len(ready_data1a)\n",
+    "# plt.pcolormesh(ready_data1[0])\n",
+    "ready_data1a[0].max().max()"
    ]
   },
   {
@@ -345,7 +346,8 @@
    "outputs": [],
    "source": [
     "for i in range(6):\n",
-    "    plt.pcolormesh(ready_data1a[i])\n",
+    "    plt.pcolormesh(ready_data1a[i], cmap=\"jet\", vmax=0.03, vmin=0.0)\n",
+    "    plt.colorbar() \n",
     "    plt.title(f'STFT Magnitude for case {i} sensor 1')\n",
     "    plt.xlabel(f'Frequency [Hz]')\n",
     "    plt.ylabel(f'Time [sec]')\n",