1 files changed, 3 insertions, 2 deletions
diff --git a/src/puzzles/puzzle.h b/src/puzzles/puzzle.h
index a65ae6f..cea0dba 100644
--- a/src/puzzles/puzzle.h
+++ b/src/puzzles/puzzle.h
@@ -100,16 +100,17 @@ typedef enum {
 } CollectionElementType;
 
 typedef struct CollectionElement {
+    Texture2D icon;
     char *path;
     char *name;
     struct CollectionElement *children;
     size_t childCount;
-    Texture2D *icon;
     bool IsInitialized;
+    bool hasIcon;
     CollectionElementType type;
 } CollectionElement;
 
 void InitializeCollection(CollectionElement *el);
-void DrawCollectionElement(CollectionElement *el, Rectangle rec);
+void DrawCollectionElement(CollectionElement *el, CollectionElement *parent, Rectangle rec);
 
 #endif // PUZZLE_H