diff --git a/etl/transform/types.py b/etl/transform/types.py
index ebccc5f..bb8b027 100644
--- a/etl/transform/types.py
+++ b/etl/transform/types.py
@@ -1,7 +1,11 @@
-from pydantic import BaseModel, constr, field_validator, EmailStr
-from typing import Optional, List
-import re
+from sqlmodel import Field, SQLModel, create_engine, Relationship
+from uuid import uuid4
 from datetime import datetime
+from pydantic import field_validator, EmailStr
+from typing import Optional, List
+
+class BaseModel(SQLModel):
+    id: str = Field(default_factory=uuid4, primary_key=True, index=True)
 
 class Dimension(BaseModel):
     floor_area_m2: float