diff --git a/NodeAPI/index.js b/NodeAPI/index.js
index d4433edda16698a43edcae142ad5b138065cff1a..46e57cc3e68bd5b4d6832345b5a567432df83eef 100644
--- a/NodeAPI/index.js
+++ b/NodeAPI/index.js
@@ -36,14 +36,22 @@ app.post('/predict', (req,res) => {
         console.log(JSON.parse(body).predictionText)
         // res.send(JSON.parse(JSON.parse(body).prediction)); //Display the response on the website
 
-            //Add to db
         // Connect to the db
         MongoClient.connect(url, function(err, db) {
-          if (err) throw err;
-          db.collection('Predictions').insertOne({
-            Plot: req.body.plot,
-            Prediction: JSON.parse(body).predictionText
-          })
+          if(err) { return console.dir(err); }
+
+          // Create collection if does not exist
+          var collection = db.createCollection('Pred_Records');
+
+          // Get the data to be stored
+          var plot = req.body.plot;
+          var prediction = JSON.parse(body).predictionText;
+
+          // Get rcord to be inserted
+          var rec = {'Plot':plot, 'Prediction': prediction};
+          
+          // Insert the record to the collection, throw error if insert unsuccessful
+          collection.insert(rec, {w:1}, function(err, result) {});
         });
 
         res.render('predictions',{predictions:JSON.parse(JSON.parse(body).predictionText)})