diff --git a/common/research/common/scripts/create_tensorflow_records.py b/common/research/common/scripts/create_tensorflow_records.py
index 9227561a0f592ed8057a20f6e46ce2df0cfd064f..9ca8456ae221659545d270e34a418205bcb4c305 100644
--- a/common/research/common/scripts/create_tensorflow_records.py
+++ b/common/research/common/scripts/create_tensorflow_records.py
@@ -12,9 +12,7 @@ def create_one_record_per_roco_dset():
         TensorflowRecordFactory.from_dataset(roco_set)
 
 
-if __name__ == "__main__":
-    # create_one_record_per_roco_dset()
-
+def create_twitch_records():
     TensorflowRecordFactory.from_dataset(
         UnionDataset(
             TwitchROCODataset.TWITCH_470149568,
@@ -24,21 +22,31 @@ if __name__ == "__main__":
             TwitchROCODataset.TWITCH_470152730,
         )
     )
-
     TensorflowRecordFactory.from_dataset(
         UnionDataset(
             TwitchROCODataset.TWITCH_470152838, TwitchROCODataset.TWITCH_470153081, TwitchROCODataset.TWITCH_470158483,
         )
     )
 
+
+def create_dji_records():
     TensorflowRecordFactory.from_dataset(
         UnionDataset(DJIROCODataset.CentralChina, DJIROCODataset.NorthChina, DJIROCODataset.SouthChina)
     )
     TensorflowRecordFactory.from_dataset(DJIROCODataset.Final)
 
+
+def create_dji_zoomed_records():
     TensorflowRecordFactory.from_dataset(
         UnionDataset(
             DJIROCOZoomedDataset.CentralChina, DJIROCOZoomedDataset.NorthChina, DJIROCOZoomedDataset.SouthChina
         )
     )
     TensorflowRecordFactory.from_dataset(DJIROCOZoomedDataset.Final)
+
+
+if __name__ == "__main__":
+    create_one_record_per_roco_dset()
+    create_twitch_records()
+    create_dji_records()
+    create_dji_zoomed_records()