diff --git a/frontend/src/lib/components/add_button.svelte b/frontend/src/lib/components/add_button.svelte
index b95464b..9de39b1 100644
--- a/frontend/src/lib/components/add_button.svelte
+++ b/frontend/src/lib/components/add_button.svelte
@@ -1,8 +1,15 @@
-
+{#if type == "button"}
+
+{:else if type == "image"}
+
+
+{:else}
+ TODO
+{/if}