diff --git a/UNet/grain_ numbers.ipynb b/UNet/grain_ numbers.ipynb
index f57fbc3414a7c6185be07c7a5eb6d191fabdf8e6..9efd501e05b8ae4973885f3bb0aebb4e0d0330e8 100644
--- a/UNet/grain_ numbers.ipynb	
+++ b/UNet/grain_ numbers.ipynb	
@@ -27,7 +27,7 @@
     "    input,_ = data[index]\n",
     "    input = copy.copy(input)\n",
     "    input = input.detach().numpy()\n",
-    "    p_mart[index]= np.sum(input[4,:,:,:])/input.size\n",
+    "    p_mart[index]= np.sum(input[4,:,:,:])/(input.size/6)\n",
     "\n",
     "  \n",
     "\n"
@@ -35,16 +35,23 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Minimum Mart percentage is:0.0248870849609375\n",
-      "Maximum Mart percentage is:0.060089111328125\n",
-      "Mean Mart percentage is:0.04008708634526112\n"
+      "Minimum Mart percentage is:0.149322509765625\n",
+      "Maximum Mart percentage is:0.36053466796875\n",
+      "Mean Mart percentage is:0.24052251807156674\n"
      ]
     }
    ],
diff --git a/UNet/visualize_vtk.ipynb b/UNet/visualize_vtk.ipynb
index e29368797afd1fac820a3da3eb735e82e0fbe004..80e7ecdac710fe17eb2d011d072352cf18204fe3 100644
--- a/UNet/visualize_vtk.ipynb
+++ b/UNet/visualize_vtk.ipynb
@@ -412,7 +412,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.9.10"
+   "version": "3.9.5"
   },
   "orig_nbformat": 4
  },