diff --git a/mnist.ipynb b/mnist.ipynb
index a4d6037..1315c1c 100644
--- a/mnist.ipynb
+++ b/mnist.ipynb
@@ -210,24 +210,12 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 21,
+   "execution_count": 6,
    "metadata": {},
-   "outputs": [
-    {
-     "ename": "OSError",
-     "evalue": "[Errno 30] Read-only file system: './linear_classifier.pkl'",
-     "output_type": "error",
-     "traceback": [
-      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
-      "\u001b[0;31mOSError\u001b[0m                                   Traceback (most recent call last)",
-      "\u001b[1;32m/arun477/vae/mnist.ipynb Cell 9\u001b[0m in \u001b[0;36m<cell line: 1>\u001b[0;34m()\u001b[0m\n\u001b[0;32m----> <a href='vscode-notebook-cell://github/arun477/vae/mnist.ipynb#X46sdnNjb2RlLXZmcw%3D%3D?line=0'>1</a>\u001b[0m \u001b[39mwith\u001b[39;00m \u001b[39mopen\u001b[39;49m(\u001b[39m'\u001b[39;49m\u001b[39m./linear_classifier.pkl\u001b[39;49m\u001b[39m'\u001b[39;49m, \u001b[39m'\u001b[39;49m\u001b[39mwb\u001b[39;49m\u001b[39m'\u001b[39;49m) \u001b[39mas\u001b[39;00m model_file:\n\u001b[1;32m      <a href='vscode-notebook-cell://github/arun477/vae/mnist.ipynb#X46sdnNjb2RlLXZmcw%3D%3D?line=1'>2</a>\u001b[0m     pickle\u001b[39m.\u001b[39mdump(model, model_file)\n",
-      "\u001b[0;31mOSError\u001b[0m: [Errno 30] Read-only file system: './linear_classifier.pkl'"
-     ]
-    }
-   ],
+   "outputs": [],
    "source": [
-    "with open('./linear_classifier.pkl', 'wb') as model_file:\n",
-    "    pickle.dump(model, model_file)"
+    "# with open('./linear_classifier.pkl', 'wb') as model_file:\n",
+    "#     pickle.dump(model, model_file)"
    ]
   },
   {
@@ -246,7 +234,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": 8,
    "metadata": {},
    "outputs": [
     {
@@ -259,15 +247,15 @@
        "        background-color: #FFFFFF;\n",
        "    }\n",
        "    #capture-button {\n",
-       "        background-color: #0074d9; /* Blue background color */\n",
-       "        color: white; /* White text color */\n",
+       "        background-color: #3F52D9; \n",
+       "        color: white;\n",
        "        border: none;\n",
        "        padding: 10px 20px;\n",
        "        cursor: pointer;\n",
        "        font-size: 16px;\n",
        "        border-radius: 3px;\n",
        "        margin-top: 10px;\n",
-       "        width: 200px;\n",
+       "        width: 400px;\n",
        "    }\n",
        "    #container {\n",
        "      display: flex;\n",
@@ -304,7 +292,7 @@
        "    });\n",
        "   var captureButton = document.getElementById('capture-button');\n",
        "    captureButton.addEventListener('click', function () {\n",
-       "      console.log('here....')\n",
+       "      // need to image save option\n",
        "    });\n",
        "</script>\n"
       ],
@@ -325,15 +313,15 @@
     "        background-color: #FFFFFF;\n",
     "    }\n",
     "    #capture-button {\n",
-    "        background-color: #0074d9; /* Blue background color */\n",
-    "        color: white; /* White text color */\n",
+    "        background-color: #3F52D9; \n",
+    "        color: white;\n",
     "        border: none;\n",
     "        padding: 10px 20px;\n",
     "        cursor: pointer;\n",
     "        font-size: 16px;\n",
     "        border-radius: 3px;\n",
     "        margin-top: 10px;\n",
-    "        width: 200px;\n",
+    "        width: 400px;\n",
     "    }\n",
     "    #container {\n",
     "      display: flex;\n",