diff --git a/python_api/api.py b/python_api/api.py
index 08595cb..f9cc6ce 100644
--- a/python_api/api.py
+++ b/python_api/api.py
@@ -382,10 +382,12 @@ async def predict(request: PredictRequest):
     # experiment_id=create_expriement( cur, conn)
 
     trial= get_trial_by_model_and_input( model_id, inputs)
-    # print(trial)
+    print("-"*20,experiment_id,"-"*20)
     if not experiment_id:
+        print("+"*20,"ENTERED IF","+"*20)
         cur,conn=get_db_cur_con()
         experiment_id=create_expriement(cur, conn)
+        print("+"*20,"ENTERED IF", experiment_id,"+"*20)
 
 
     # print(trail)
@@ -401,7 +403,7 @@ async def predict(request: PredictRequest):
         new_trial_id=create_trial( model_id, experiment_id, cur, conn,source_trial)
         # if not experiment_id:
         #     experiment_id=create_expriement(cur, conn)
-
+        print("*"*20,"RETURNING IF TRIAL", experiment_id,"*"*20)
         return {"experimentId": experiment_id, "trialId": new_trial_id, "model_id": model["name"], "input_url": inputs}
     else:
         cur,conn=get_db_cur_con()
@@ -429,6 +431,7 @@ async def predict(request: PredictRequest):
         message= makePredictMessage(architecture, batch_size, desired_result_modality, gpu, inputs,has_multi_input,context,config, model["name"], trace_level, 0, "localhost:6831")
 
         sendPredictMessage(message,queue_name,trial_id)
+        print("*"*20,"RETURNING ELSE TRIAL", experiment_id,"*"*20)
         return {"experimentId": experiment_id, "trialId": trial_id, "model_id": model["name"],"input_url": inputs}
 
 
diff --git a/python_api/db.py b/python_api/db.py
index 432927d..936f82e 100644
--- a/python_api/db.py
+++ b/python_api/db.py
@@ -45,7 +45,7 @@ def create_trial_inputs(trial_id, inputs, cur, conn):
     while True:
         try:
             # Fetch the latest max ID
-            cur.execute("SELECT LAST_VALUE(id) OVER (ORDER BY id ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING) AS id FROM trial_inputs LIMIT 1")
+            cur.execute("SELECT MAX(id) as id FROM trial_inputs")
             max_id = cur.fetchone()["id"]
             max_id = int(max_id) if max_id is not None else 0  # Handle NULL case