1 files changed, 8 insertions, 2 deletions
diff --git a/src/state.h b/src/state.h
index 696b657..38cc63f 100644
--- a/src/state.h
+++ b/src/state.h
@@ -1,18 +1,24 @@
 #ifndef COMET_STATE_H
 #define COMET_STATE_H
 
+#include <glib.h>
+
 #include "window.h"
 #include "draw.h"
+#include "button.h"
 
 typedef struct State State;
 
 struct State {
-	Window *win;
-	Drawable *draw;
+    Window *win;
+    Drawable *draw;
+    GList *btns;
 };
 
 State *state_create(Window *win, Drawable *draw);
 
+void state_add_button(State *state, Button *btn);
+
 void state_destroy(State *state);
 
 #endif