diff --git a/test-wemos.yaml b/test-wemos.yaml
index d75ac4a..4669e55 100644
--- a/test-wemos.yaml
+++ b/test-wemos.yaml
@@ -1,3 +1,8 @@
+substitutions:
+  # Modify variables based on your settings
+  hostname: "test-wemos"
+  devicename: test-wemos
+
 esphome:
   name: test-wemos
   friendly_name: test_wemos
@@ -28,74 +33,23 @@ wifi:
     subnet: 255.255.255.0
   power_save_mode: none
 
-  # Enable fallback hotspot (captive portal) in case wifi connection fails
-  # ap:
-  #   ssid: "Esp32-C3-Mini Fallback Hotspot"
-  #   password: "wuMGlIsDXN19"
-
-# one_wire:
-#   - platform: gpio
-#     pin: GPIO4
-#     id: bus1
-
-# i2c:
-#   - id: bus_a
-#     sda: GPIO6
-#     scl: GPIO7
-#     scan: true
 
 
 
 captive_portal:
 
 sensor:
-  # - platform: ina219
-  #   address: 0x40
-  #   shunt_resistance: 0.1 ohm
-  #   current:
-  #     name: "INA219 Current"
-  #     unit_of_measurement: "A"
-  #   power:
-  #     name: "INA219 Power"
-  #     unit_of_measurement: "W"
-  #     filters:
-  #     - multiply: 10
-  #   bus_voltage:
-  #     name: "INA219 Bus Voltage"
-  #     unit_of_measurement: "V"
-  #   shunt_voltage:
-  #     name: "INA219 Shunt Voltage"
-  #     unit_of_measurement: "V"
-  #   max_voltage: 32.0V
-  #   max_current: 3.2A
-  #   update_interval: 1s
-
-  # - platform: dallas_temp
-  #   one_wire_id: bus1
-  #   name: "DS18B20"
-  #   update_interval: 2s
-
-  # - platform: bme680
-  #   i2c_id: bus_a
-  #   address: 0x76
-  #   update_interval: 1s
-
-# - platform: gpio
-#     id: switch_output1
-#     name: "a8-output1"
-#     pin:
-#       pcf8574: pcf8574_hub_out_1
-#       number: 0
-#       mode: OUTPUT
-#       inverted: true
-
 
 switch:
   - platform: gpio
+    name: switch
     id: switch_output1
     pin: GPIO4
     inverted: True
+    restore_mode: ALWAYS_OFF
     
+  - platform: restart
+    name: ${devicename} restart
   
 web_server:
   port: 80