diff --git a/alarm-panel-pro-esp32-wifi.yaml b/alarm-panel-pro-esp32-wifi.yaml
index 2b35599..5281d43 100644
--- a/alarm-panel-pro-esp32-wifi.yaml
+++ b/alarm-panel-pro-esp32-wifi.yaml
@@ -78,7 +78,7 @@ packages:
       # Enables a 'Warning Beep' entity, intended to be used with a piezo buzzer. This is 
       # implemented using a light entity with strobe effect to create a repeated beeping sound.
       - packages/warning-beep.yaml
-  
+
 dashboard_import:
   package_import_url: github://konnected-io/konnected-esphome/alarm-panel-pro-esp32-wifi.yaml@master
   import_full_config: false
diff --git a/packages/alarm-panel-esp32-base.yaml b/packages/alarm-panel-esp32-base.yaml
index c0d07b4..0834cd6 100644
--- a/packages/alarm-panel-esp32-base.yaml
+++ b/packages/alarm-panel-esp32-base.yaml
@@ -12,7 +12,7 @@ substitutions:
   name: alarm-panel-pro
   friendly_name: Alarm Panel Pro
   project_name: konnected.alarm-panel-pro-esp32-wifi
-  project_version: "0.3.2"
+  project_version: "0.4.0"
 
   sensor_debounce_time: 200ms
   warning_beep_pulse_time: 100ms
diff --git a/packages/core-esp32.yaml b/packages/core-esp32.yaml
index ef24c77..d04f1cc 100644
--- a/packages/core-esp32.yaml
+++ b/packages/core-esp32.yaml
@@ -14,7 +14,7 @@ esphome:
 esp32:  
   board: wesp32
   framework:
-    type: arduino
+    type: esp-idf
     version: recommended
 
 substitutions:
@@ -32,6 +32,10 @@ sensor:
     name: Uptime
     entity_category: diagnostic
 
-switch:
+button:
   - platform: restart
-    name: Restart
\ No newline at end of file
+    name: Restart
+    entity_category: config
+  # - platform: factory_reset
+  #   name: Factory Reset
+  #   entity_category: config
\ No newline at end of file
diff --git a/packages/wifi-esp32.yaml b/packages/wifi-esp32.yaml
index 06a6ca3..a40e2fb 100644
--- a/packages/wifi-esp32.yaml
+++ b/packages/wifi-esp32.yaml
@@ -5,8 +5,8 @@ wifi:
 improv_serial:
 captive_portal:
 
-# esp32_improv:
-#   authorizer: false
+esp32_improv:
+  authorizer: false
 
 sensor:
   - platform: wifi_signal