1 files changed, 8 insertions, 0 deletions
diff --git a/src/level.h b/src/level.h
index 897daa5..e915da3 100644
--- a/src/level.h
+++ b/src/level.h
@@ -3,6 +3,9 @@
 #include "bitmap.h"
 #include <cstdint>
 #include <cstring>
+#include <vector>
+
+class Train;
 
 class Level {
 public:
@@ -13,13 +16,18 @@ public:
 	void set(int x, int y, uint8_t tile);
 	bool inBounds(int x, int y) const { return x >= 0 && x < m_width && y >= 0 && y < m_height; }
 
+	void update();
 	void draw(Bitmap& bitmap, int xo, int yo);
+
+	void addVehicle(Train&);
 	void toggleTile(int x, int y);
 
 private:
 	Bitmap& m_tileSprites;
 	int m_width, m_height;
 	uint8_t* m_tiles;
+
+	std::vector<Train*> m_vehicles;
 };
 
 enum RailDirection {