diff --git a/cognee/infrastructure/engine/models/DataPoint.py b/cognee/infrastructure/engine/models/DataPoint.py
index 9ab8cf5ca..9e504be9a 100644
--- a/cognee/infrastructure/engine/models/DataPoint.py
+++ b/cognee/infrastructure/engine/models/DataPoint.py
@@ -25,6 +25,11 @@ class DataPoint(BaseModel):
     version: int = 1  # Default version
     topological_rank: Optional[int] = 0
     metadata: Optional[MetaData] = {"index_fields": []}
+    type: str = Field(default_factory=lambda: DataPoint.__name__)
+
+    def __init__(self, **data):
+        super().__init__(**data)
+        object.__setattr__(self, "type", self.__class__.__name__)
 
     @classmethod
     def get_embeddable_data(self, data_point: "DataPoint"):
diff --git a/cognee/modules/visualization/cognee_network_visualization.py b/cognee/modules/visualization/cognee_network_visualization.py
index 24c8e1f66..cf4162044 100644
--- a/cognee/modules/visualization/cognee_network_visualization.py
+++ b/cognee/modules/visualization/cognee_network_visualization.py
@@ -19,6 +19,7 @@ async def cognee_network_visualization(graph_data, destination_file_path: str =
         "Entity": "#f47710",
         "EntityType": "#6510f4",
         "DocumentChunk": "#801212",
+        "TextSummary": "#1077f4",
         "default": "#D3D3D3",
     }