1 files changed, 5 insertions, 1 deletions
diff --git a/src/game/GameModel.h b/src/game/GameModel.h
index b8f5217..72e2875 100644
--- a/src/game/GameModel.h
+++ b/src/game/GameModel.h
@@ -25,7 +25,8 @@ private:
 	vector<Tool*> toolList;
 	vector<Menu*> menuList;
 	Menu * activeMenu;
-	Brush * currentBrush;
+	int currentBrush;
+	vector<Brush *> brushList;
 	Save * currentSave;
 	Simulation * sim;
 	Renderer * ren;
@@ -43,6 +44,7 @@ private:
 public:
 	GameModel();
 	~GameModel();
+
 	Save * GetSave();
 	Brush * GetBrush();
 	void SetSave(Save * newSave);
@@ -58,6 +60,8 @@ public:
 	Menu * GetActiveMenu();
 	User GetUser();
 	void SetUser(User user);
+	void SetBrush(int i);
+	int GetBrushID();
 	Simulation * GetSimulation();
 	Renderer * GetRenderer();
 };