diff --git a/controllers/train.go b/controllers/train.go
index cd73558..42429ff 100644
--- a/controllers/train.go
+++ b/controllers/train.go
@@ -73,7 +73,10 @@ func (tc *TrainingController) CreateTrainingRecord(c *gin.Context) {
 		return
 	}
 
-	c.JSON(http.StatusCreated, record)
+	c.JSON(http.StatusCreated, gin.H{
+		"message": "Data saved successfully",
+		"id":      record.TrainId,
+	})
 }
 
 func ReceiveTrainingData(c *gin.Context) {