diff --git a/nocodetools/models.py b/nocodetools/models.py
index 0213803..f4d86bc 100644
--- a/nocodetools/models.py
+++ b/nocodetools/models.py
@@ -1,14 +1,25 @@
 from django.db import models
+from django.utils import timezone
 
 import mongoengine as me
 
+
 class Element(me.EmbeddedDocument):
+    element_id = me.StringField()
     element_type = me.StringField()
     content = me.StringField()
     css = me.DictField()
 
-class Page(me.Document):
+class Page(me.EmbeddedDocument):
     cut = me.IntField()
     elements = me.ListField(me.EmbeddedDocumentField(Element))
-    created_at = me.DateTimeField()
-    updated_at = me.DateTimeField()
\ No newline at end of file
+
+class Code(me.Document):
+    pages = me.ListField(me.EmbeddedDocumentField(Page))
+    created_at = me.DateTimeField(default=timezone.now)
+    updated_at = me.DateTimeField(default=timezone.now)
+
+    def save(self, *args, **kwargs):
+        self.updated_at = timezone.now()
+        return super().save(*args, **kwargs)
+    
\ No newline at end of file