diff --git a/scripts/06_generate_dataset.py b/scripts/06_generate_dataset.py
index d3fb72fe9f009a739d7ac385f8b1914b55f294d3..79f8dc400b01d975a510c06a29ad65664650d2b3 100644
--- a/scripts/06_generate_dataset.py
+++ b/scripts/06_generate_dataset.py
@@ -11,7 +11,9 @@ and needs 100gb of space.
 
 def main():
     # Dataset save folder
-    save_folder = Path(r'C:\data\XrayTransformLow')
+    save_folder = Path(r'\\192.168.43.8\data\Wittl\06_datasets\XrayTransform')
+    number_of_maps = 4
+    map_bounding_box = 25
     # Dataset of multiple random stl files from: https://ten-thousand-models.appspot.com/
     load_folder = Path(r'C:\Users\swittl\Downloads\Thingi10K\Thingi10K\raw_meshes')
     
@@ -22,7 +24,7 @@ def main():
         print(i)
         key = random.PRNGKey(i)    
         generate_atlas_from_folder(
-            load_folder, key, save_folder, map_bounding_box=(-10, 10), number_of_maps=2)
+            load_folder, key, save_folder, map_bounding_box=(-map_bounding_box, map_bounding_box), number_of_maps=number_of_maps)
     
 
 if __name__ == '__main__':