diff --git a/scripts/runSimulation.py b/scripts/runSimulation.py
index 1b124d6ff86eb382bc485c7eb469859b64014a0a..d5db5d4ec1823353a3118519b109c095f95bfd3e 100644
--- a/scripts/runSimulation.py
+++ b/scripts/runSimulation.py
@@ -1,3 +1,5 @@
+#--------------------------- Run SOFA Simulation - Saurabh Deshpande -----------------------------#
+
 import os
 import subprocess
 import numpy as np
@@ -6,7 +8,7 @@ import pandas as pd
 
 
 #Or we can generate forces in random directions 
-
+#qt_nodes.csv contains all the nodes on which forces will be to generate the training data 
 df = pd.read_csv('qt_nodes.csv', names=['u'], header = None)
 topnodes= df['u'].values
 
@@ -15,7 +17,7 @@ topnodes = [ int(x) for x in topnodes]
 #print(topnodes)
 #print("Length of array is", len(topnodes))
 #number of training example generated at a particular node 
-m = 450
+m = 100
 
 #Index on which forces are applied on this node numbers 
 index = topnodes