diff --git a/Project_Model/UML/model.excalidraw.json b/Project_Model/UML/model.excalidraw.json
index 5bdbe5e..3eef683 100644
--- a/Project_Model/UML/model.excalidraw.json
+++ b/Project_Model/UML/model.excalidraw.json
@@ -5933,6 +5933,80 @@
       "fixedSegments": null,
       "startIsSpecial": null,
       "endIsSpecial": null
+    },
+    {
+      "id": "kCkl_cGFGtvEjte0RLULg",
+      "type": "text",
+      "x": 2852.241705006319,
+      "y": 1434.6417703971483,
+      "width": 197.99972534179688,
+      "height": 50,
+      "angle": 0,
+      "strokeColor": "#1e1e1e",
+      "backgroundColor": "transparent",
+      "fillStyle": "solid",
+      "strokeWidth": 2,
+      "strokeStyle": "solid",
+      "roughness": 1,
+      "opacity": 100,
+      "groupIds": [],
+      "frameId": null,
+      "index": "b37",
+      "roundness": null,
+      "seed": 1464660035,
+      "version": 158,
+      "versionNonce": 220662563,
+      "isDeleted": false,
+      "boundElements": null,
+      "updated": 1759673320534,
+      "link": null,
+      "locked": false,
+      "text": "class AttentioMask\nbbgbgd",
+      "fontSize": 20,
+      "fontFamily": 8,
+      "textAlign": "left",
+      "verticalAlign": "top",
+      "containerId": null,
+      "originalText": "class AttentioMask\nbbgbgd",
+      "autoResize": true,
+      "lineHeight": 1.25
+    },
+    {
+      "id": "cazVvK8SPLZZI6PecpGbM",
+      "type": "text",
+      "x": 3450.385626322292,
+      "y": 1363.0943062968788,
+      "width": 703.9990844726562,
+      "height": 250,
+      "angle": 0,
+      "strokeColor": "#1e1e1e",
+      "backgroundColor": "transparent",
+      "fillStyle": "solid",
+      "strokeWidth": 2,
+      "strokeStyle": "solid",
+      "roughness": 1,
+      "opacity": 100,
+      "groupIds": [],
+      "frameId": null,
+      "index": "b38",
+      "roundness": null,
+      "seed": 1191002637,
+      "version": 636,
+      "versionNonce": 1757202541,
+      "isDeleted": false,
+      "boundElements": null,
+      "updated": 1759674426339,
+      "link": null,
+      "locked": false,
+      "text": "class NanoSocratesCore:\n    # manages model\n    - encoder_block: nn.Sequential\n    - decoder_block: nn.Sequential\n\n    + forward(x: torch.Tensor, taskType: TaskType): torch.Tensor\n    - full(x: torch.Tensor): torch.Tensor # 1-2\n    - encoder_only(x: torch.Tensor): torch.Tensor # 3\n    - decoder_only(x: torch.Tensor): torch.Tensor # 4\n",
+      "fontSize": 20,
+      "fontFamily": 8,
+      "textAlign": "left",
+      "verticalAlign": "top",
+      "containerId": null,
+      "originalText": "class NanoSocratesCore:\n    # manages model\n    - encoder_block: nn.Sequential\n    - decoder_block: nn.Sequential\n\n    + forward(x: torch.Tensor, taskType: TaskType): torch.Tensor\n    - full(x: torch.Tensor): torch.Tensor # 1-2\n    - encoder_only(x: torch.Tensor): torch.Tensor # 3\n    - decoder_only(x: torch.Tensor): torch.Tensor # 4\n",
+      "autoResize": true,
+      "lineHeight": 1.25
     }
   ],
   "appState": {