diff --git a/layouts/shortcodes/button.html b/layouts/shortcodes/button.html
index fb87840..7c000a3 100644
--- a/layouts/shortcodes/button.html
+++ b/layouts/shortcodes/button.html
@@ -1,5 +1,5 @@
{{- $ref := "" }}
-{{- $target := "" }}
+{{- $class := "" }}
{{- $size := default "regular" (.Get "size" | lower) }}
{{- if not (in (slice "regular" "large") $size) }}
@@ -8,19 +8,21 @@
{{- with .Get "href" }}
{{- $ref = . }}
- {{- $target = "_blank" }}
{{- end }}
{{- with .Get "relref" }}
{{- $ref = relref $ . }}
{{- end }}
+{{- with .Get "class" }}
+ {{- $class = . }}
+{{- end }}
-
+
+
{{ $.Inner }}