2 files changed, 10 insertions, 3 deletions

diff --git a/tpl/index.pug b/tpl/index.pug
index fa9bd85..510e09b 100644
--- a/tpl/index.pug
+++ b/tpl/index.pug
@@ -476,7 +476,7 @@ html
             +rule(level='medium')
 
             +box
-                div(style={ display: 'flex', gap: '2rem' })
+                div(style={ display: 'flex', gap: '3rem' })
                     +menu
                         +menu-header= 'Category 1'
                         +menu-item= 'Mark as away'
diff --git a/tpl/objects/menu.pug b/tpl/objects/menu.pug
index 7c9c07e..97a571d 100644
--- a/tpl/objects/menu.pug
+++ b/tpl/objects/menu.pug
@@ -1,12 +1,19 @@
 include icon.pug
 include status-indicator.pug
+include action-button.pug
 
 mixin menu
     .o-menu
         block
 
 mixin menu-item
-    button.o-menu__item(disabled=attributes.disabled)
+    -
+        let classes = {
+            'o-menu__item': true,
+            'is-selected':  attributes.selected,
+        }
+
+    button(class=classes, disabled=attributes.disabled)
         .l-card.l-card--flush
             if attributes.icon
                 .l-card__block.o-menu__icon-slot
@@ -14,7 +21,7 @@ mixin menu-item
                         +status-indicator(attributes.icon)(class='o-menu__icon')
                     else
                         +icon(attributes.icon)(class='o-menu__icon')
-            .l-card__block.l-card__block--main
+            .l-card__block.l-card__block--main&attributes({ class: attributes.contentClass })
                 block
             if attributes.postIcon
                 .l-card__block.o-menu__icon-slot