diff --git a/llvm/projects/soc_simulator/scripts/construct_ops.py b/llvm/projects/soc_simulator/scripts/construct_ops.py
index 3b655f2f5fb7ccb3eb4ac8db2e105cb74d71f986..3bcb2af9c345b19d86acaf92e3771c60370b4678 100644
--- a/llvm/projects/soc_simulator/scripts/construct_ops.py
+++ b/llvm/projects/soc_simulator/scripts/construct_ops.py
@@ -11,7 +11,7 @@ op_map["pool"] = "Pool"
 op_map["relu"] = "Relu"
 op_map["activation"] = "Relu"
 op_map["tanh"] = "Tanh"
-op_map["batchnorm"] = "NML"
+op_map["batchnorm"] = "BatchNorm"
 
 
 unique_op_map = {}
@@ -45,19 +45,29 @@ if __name__ == "__main__":
   f2 = open(out_path, "w+")
 
 
-
   nml_id = 1
   conv_id = 1
   fc_id = 1
+  batchnorm_id = 1
   
   for x in f:
       toks = x.split()
 
       layer_len = len(toks)
 
+      #if "batchnorm" in toks:
+      #    f2.write("BatchNorm" + str(batchnorm_id) + "\n")
+      #    batchnorm_id += 1
+      #    continue
+      
       if layer_len == 1 and "conv" not in toks and "dense" not in toks:
           f2.write("#NML" + str(nml_id) + ",1\n")
           nml_id += 1
+
+          layer_str = getLayerStr(toks)
+          f2.write(layer_str)
+      
+          
       if "conv" in toks:
           f2.write("#Conv" + str(conv_id) + "," + str(layer_len) + "\n")