diff --git a/db.py b/db.py
index 4702c3d..4c79e7c 100644
--- a/db.py
+++ b/db.py
@@ -1,10 +1,14 @@
 import os
 from sqlmodel import create_engine, SQLModel, Session
+from models.station import Station
+from models.measurement import IndoorMeasurement, OutdoorMeasurement
 
 DATA_DIR = "data"
 os.makedirs(DATA_DIR, exist_ok=True)
 
 engine = create_engine(f"sqlite:///{DATA_DIR}/database.db")
+
+
 SQLModel.metadata.create_all(engine)
 
 def get_session():