diff --git a/.gitignore b/.gitignore
index f0fa0ae..5da3adb 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,3 +1,5 @@
 .vscode/
 .devcontainer/
 build/
+example_ota_ws/sdkconfig
+example_ota_ws/sdkconfig.old
diff --git a/example_ota_ws/dependencies.lock b/example_ota_ws/dependencies.lock
new file mode 100644
index 0000000..f39ac52
--- /dev/null
+++ b/example_ota_ws/dependencies.lock
@@ -0,0 +1,9 @@
+dependencies:
+  idf:
+    component_hash: null
+    source:
+      type: idf
+    version: 5.2.0
+manifest_hash: 2ac596d0d5a708817329b21531cbf3574c7c8daaf36dca979b5dd32a78fc2857
+target: esp32s3
+version: 1.0.0
diff --git a/example_ota_ws/example_ota_ws.code-workspace b/example_ota_ws/example_ota_ws.code-workspace
new file mode 100644
index 0000000..437ac53
--- /dev/null
+++ b/example_ota_ws/example_ota_ws.code-workspace
@@ -0,0 +1,11 @@
+{
+	"folders": [
+		{
+			"path": "."
+		},
+		{
+			"path": ".."
+		}
+	],
+	"settings": {}
+}
\ No newline at end of file
diff --git a/include/ota_ws_update.h b/include/ota_ws_update.h
index fe0a1e4..30f61f0 100644
--- a/include/ota_ws_update.h
+++ b/include/ota_ws_update.h
@@ -1,7 +1,5 @@
 #pragma once
 
-#include "ota_ws_private.h"
-
 #ifdef __cplusplus
 extern "C"
 {
diff --git a/source/ota_ws_update.html b/source/ota_ws_update.html
index 552a7b4..1d00c7c 100644
--- a/source/ota_ws_update.html
+++ b/source/ota_ws_update.html
@@ -70,7 +70,7 @@
         <button class="btn" id="goHome">Home Page</button>
     </div>
     <div id="rollback" style="display:none">
-        <div class="column" >
+        <div class="column">
             <button class="btn" id="otaVerifyApp">Click to confirm and commit OTA update</button>
         </div>
         <div class="column">
@@ -115,8 +115,6 @@
 
             reader.onload = function () {
                 otaData = new Uint8Array(reader.result);
-                //                console.log(reader.result);
-                //                console.log(otaData.length);
                 document.getElementById("otaStartVisible").style.display = "block";
                 document.getElementById("otaProgressVisible").style.display = "none";
                 document.getElementById("otaReStartVisible").style.display = "none";
@@ -132,18 +130,14 @@
             if (otaData.length > 0 && otaStarted == 0) {
 
                 socket.send(JSON.stringify({ name: "otaSize", value: otaData.length }));
-                //console.log(JSON.stringify({ name: "otaSize", value: otaData.length }));
-
                 otaStarted = 1;
                 this.innerHTML = "Click to Cancel";
                 document.getElementById("otaFileSelect").disabled = true;
                 document.getElementById("otaProgressVisible").style.display = "block";
                 document.getElementById("otaPogress").max = otaData.length;
-                //tstReceive();
             }
             else {
                 otaStarted = 0;
-                //receiveWsData(JSON.stringify({ name: "otaCancel", value: "Cancel" }));
                 socket.send(JSON.stringify({ name: "otaCancel", value: "Cancel" }));
             }
 
@@ -155,10 +149,9 @@
         });
         document.getElementById("otaReStart").addEventListener("click", function (e) {
             socket.send(JSON.stringify({ name: "otaRestartEsp", value: "restart" }));
-            //console.log(JSON.stringify({ name: "otaRestartEsp", value: "restart" }));
         });
 
-        /*
+        /*  // debug -> test sample
         function sleep(ms) {
             return new Promise(resolve => setTimeout(resolve, ms));
         }
@@ -171,7 +164,6 @@
                 }
                 otaStartsegment += otaSetChunkSize;
             }
-            //console.log(otaStartsegment + " " + otaSetChunkSize + " " + otaData.length + " " + (otaData.length - otaStartsegment));
             if (otaStarted == 1) {
                 receiveWsData(JSON.stringify({ name: "otaGetChunk", value: otaStartsegment }));
                 receiveWsData(JSON.stringify({ name: "otaEnd", value: "OK" }));
@@ -181,7 +173,6 @@
         function receiveWsData(data) {
             try {
                 let obj = JSON.parse(data);
-                //console.log(data);
                 switch (obj.name) {
                     case "otaSetChunkSize":
                         otaSetChunkSize = obj.value;
@@ -190,7 +181,6 @@
                         let otaDataSend = otaData.subarray(obj.value, obj.value + otaSetChunkSize);
                         document.getElementById("otaPogress").value = obj.value;
                         document.getElementById("otaStartCancel").innerHTML = "Ota download. Size = " + otaData.length + " Segment = " + obj.value + " Click to Cancel";
-                        //console.log("sock send " + obj.value + " " + otaDataSend.length);
                         socket.send(otaDataSend);
                         break;
                     case "otaEnd":
@@ -224,24 +214,22 @@
             }
             catch
             {
-                console.log(data + " catch");
+                console.log(data + "Error msg");
             }
         };
     </script>
 
     <script>  // rollback
         document.getElementById("otaVerifyApp").addEventListener("click", function (e) {
-        socket.send(JSON.stringify({ name: "otaProcessRollback", value: "false" }));
-        document.getElementById("rollback").style.display = "none";
-        document.getElementById("update").style.display = "block";
-            console.log(JSON.stringify({ name: "otaProcessRollback", value: "false" }));
+            socket.send(JSON.stringify({ name: "otaProcessRollback", value: "false" }));
+            document.getElementById("rollback").style.display = "none";
+            document.getElementById("update").style.display = "block";
         });
 
         document.getElementById("otaRollback").addEventListener("click", function (e) {
-        socket.send(JSON.stringify({ name: "otaProcessRollback", value: "true" }));
-        document.getElementById("rollback").style.display = "none";
-        document.getElementById("update").style.display = "block";
-            console.log(JSON.stringify({ name: "otaProcessRollback", value: "true" }));
+            socket.send(JSON.stringify({ name: "otaProcessRollback", value: "true" }));
+            document.getElementById("rollback").style.display = "none";
+            document.getElementById("update").style.display = "block";
         });
 
     </script>
@@ -256,14 +244,15 @@
 
     <script> // события WS
         socket.onopen = function () {
-            console.log("connect");
+            console.log("connect ws");
         };
         socket.onclose = function (event) {
-            console.log("close");
+            document.location.reload();
+            console.log("close ws");
         };
         socket.onerror = function () {
             document.location.reload();
-            console.log("error");
+            console.log("error ws");
         };
         socket.onmessage = function (event) {
             receiveWsData(event.data);
diff --git a/source/ota_ws_update_esp.c b/source/ota_ws_update_esp.c
index a5d0f6e..df88d38 100644
--- a/source/ota_ws_update_esp.c
+++ b/source/ota_ws_update_esp.c
@@ -10,7 +10,7 @@
 #include "esp_partition.h"
 #include "esp_image_format.h"
 
-#include "ota_ws_private.h"
+#include "ota_ws_update_private.h"
 
 static const char *TAG = "ota_ws_esp";
 
@@ -27,6 +27,11 @@ esp_err_t start_ota_ws(void)
 
     const esp_partition_t *configured = esp_ota_get_boot_partition();
     const esp_partition_t *running = esp_ota_get_running_partition();
+    if(configured==NULL || running == NULL)
+    {
+        ESP_LOGE(TAG,"OTA data not found");
+        return ESP_FAIL;
+    }
 
     if (configured != running)
     {
diff --git a/source/ota_ws_update_http.c b/source/ota_ws_update_http.c
index 57815c5..5af4e5e 100644
--- a/source/ota_ws_update_http.c
+++ b/source/ota_ws_update_http.c
@@ -5,8 +5,8 @@
    software is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
    CONDITIONS OF ANY KIND, either express or implied.
 */
-#include "ota_ws_private.h"
-#include "ota_ws.h"
+#include "ota_ws_update_private.h"
+#include "ota_ws_update.h"
 
 #include "freertos/task.h"
 #include "freertos/queue.h"
@@ -234,11 +234,11 @@ _recv_ret:
 // send http initial page and js code
 static esp_err_t ota_get_handler(httpd_req_t *req)
 {
-    extern const unsigned char ota_ws_html_start[] asm("_binary_ota_ws_html_start");
-    extern const unsigned char ota_ws_html_end[] asm("_binary_ota_ws_html_end");
-    const size_t ota_ws_html_size = (ota_ws_html_end - ota_ws_html_start);
+    extern const unsigned char ota_ws_update_html_start[] asm("_binary_ota_ws_update_html_start");
+    extern const unsigned char ota_ws_update_html_end[] asm("_binary_ota_ws_update_html_end");
+    const size_t ota_ws_update_html_size = (ota_ws_update_html_end - ota_ws_update_html_start);
 
-    httpd_resp_send_chunk(req, (const char *)ota_ws_html_start, ota_ws_html_size);
+    httpd_resp_send_chunk(req, (const char *)ota_ws_update_html_start, ota_ws_update_html_size);
     httpd_resp_sendstr_chunk(req, NULL);
     return ESP_OK;
 }