diff --git a/main/ota_update.c b/main/ota_update.c
index 44feb32..c0a0fe2 100644
--- a/main/ota_update.c
+++ b/main/ota_update.c
@@ -94,7 +94,7 @@ void start_uart_update(uint8_t msgid, const uint8_t *payload,
   uart_write_bytes(MASTER_UART, send_buffer, len);
 }
 
-esp_err_t write_ota(uint32_t write_len, const uint8_t *payload) {
+esp_err_t write_ota_update(uint32_t write_len, const uint8_t *payload) {
   if (update_buffer_write_index > UPDATE_BUFFER_SIZE - write_len) {
     // ESP_LOGI(TAG, "Writing Data to Update BUffer Sequence %d, writing Data
     // %d",
@@ -124,7 +124,7 @@ void payload_uart_update(uint8_t msgid, const uint8_t *payload,
   uint32_t write_len = MIN(UPDATE_PAYLOAD_SIZE, payload_len);
   update_size += write_len;
 
-  esp_err_t err = write_ota(write_len, payload);
+  esp_err_t err = write_ota_update(write_len, payload);
 
   if (err != ESP_OK) {
     ESP_LOGE(TAG, "GOT ESP ERROR WRITE OTA %d", err);
@@ -148,7 +148,7 @@ void payload_uart_update(uint8_t msgid, const uint8_t *payload,
   uart_write_bytes(MASTER_UART, send_buffer, len);
 }
 
-esp_err_t end_ota() {
+esp_err_t end_ota_update() {
   esp_err_t err =
       esp_ota_write(update_handle, update_buffer, update_buffer_write_index);
   if (err != ESP_OK) {
@@ -185,7 +185,7 @@ void end_uart_update(uint8_t msgid, const uint8_t *payload, size_t payload_len,
                      size_t send_buffer_size) {
   ESP_LOGI(TAG, "OTA Update End Uart Command");
 
-  esp_err_t err = end_ota();
+  esp_err_t err = end_ota_update();
 
   // message ret esp_err_t
   int send_payload_len = 1;
diff --git a/main/ota_update.h b/main/ota_update.h
index f8007b3..6433486 100644
--- a/main/ota_update.h
+++ b/main/ota_update.h
@@ -1,20 +1,25 @@
 #ifndef OTA_UPDATE_H
 #define OTA_UPDATE_H
 
+#include "esp_err.h"
+#include <stdint.h>
+#include <sys/types.h>
 #define UPDATE_BUFFER_SIZE 4000
 #define UPDATE_PAYLOAD_SIZE 200
-#define UPDATE_MAX_SEQUENZES (UPDATE_BUFFER_SIZE/UPDATE_PAYLOAD_SIZE)
+#define UPDATE_MAX_SEQUENZES (UPDATE_BUFFER_SIZE / UPDATE_PAYLOAD_SIZE)
 
 void init_ota();
 
 enum OTA_UPDATE_STATES {
-	IDEL,
-	START_REQUESTED,
-	WAITING_FOR_PAYLOAD,
-	WRITING_OTA_TO_PARTITION,
+  IDEL,
+  START_REQUESTED,
+  WAITING_FOR_PAYLOAD,
+  WRITING_OTA_TO_PARTITION,
 
-	
-	
 };
 
+int prepare_ota_update();
+esp_err_t write_ota_update(uint32_t write_len, const uint8_t *payload);
+esp_err_t end_ota_update();
+
 #endif