diff --git a/hue-dimmer-blueprint.yaml b/hue-dimmer-blueprint.yaml
index 26449ed..411ada0 100644
--- a/hue-dimmer-blueprint.yaml
+++ b/hue-dimmer-blueprint.yaml
@@ -6,18 +6,26 @@ blueprint:
     switch:
         selector:
             device:
-    button1_short_press:
-        default:
-        selector:
-            action:
-    button2_long_press:
-        default:
-        selector:
-            action:
-    button1_hold:
-        default:
-        selector:
-            action:
+                filter:
+                    model: Hue dimmer switch (RWL021)
+    button1_section:
+        name: Button 1
+        collapsed: true
+        input:
+          button1_short_press:
+              default:
+              selector:
+                  action:
+        input:
+          button2_long_press:
+              default:
+              selector:
+                  action:
+        input:
+          button1_hold:
+              default:
+              selector:
+                  action:
 triggers:
   - event_type: state_changed
     event_data: