diff --git a/include/depthai-bootloader-shared/Bootloader.hpp b/include/depthai-bootloader-shared/Bootloader.hpp
index 034c6f0..a22ffbc 100644
--- a/include/depthai-bootloader-shared/Bootloader.hpp
+++ b/include/depthai-bootloader-shared/Bootloader.hpp
@@ -21,6 +21,9 @@ namespace request {
         // Common
         Command cmd;
         UsbRomBoot() : cmd(USB_ROM_BOOT) {}
+
+        // Data
+        bool keepUsbBootAfterAppRestart = false;
     };
     struct BootApplication {
         // Common
@@ -36,8 +39,12 @@ namespace request {
         UpdateFlash() : cmd(UPDATE_FLASH) {}
 
         // Data
-        enum Storage : uint32_t { SBR, BOOTLOADER };
+        enum Storage : uint32_t { SBR, BOOTLOADER, BOOT_HEADER };
+        // TODO make BOOT_HEADER and BOOTLOADER storage not overlap.
+        // Then we could have separate actions to update BOOT_HEADER only.
+        enum BootHeader : uint32_t { CUSTOM_PAYLOAD, USB, /* FLASH_SPI_QUAD, */ };
         Storage storage;
+        BootHeader bootHeader;
         uint32_t totalSize;
         uint32_t numPackets;
     };