diff --git a/hydrogym/experimental/__init__.py b/hydrogym/experimental/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/hydrogym/core_1DEnvs.py b/hydrogym/experimental/core_1DEnvs.py
similarity index 94%
rename from hydrogym/core_1DEnvs.py
rename to hydrogym/experimental/core_1DEnvs.py
index 92730df..2f60824 100644
--- a/hydrogym/core_1DEnvs.py
+++ b/hydrogym/experimental/core_1DEnvs.py
@@ -136,7 +136,6 @@ def evaluate_objective(self) -> ArrayLike:
     pass
 
 if __name__ == '__main__':
-    # torch.set_default_tensor_type('torch.cuda.FloatTensor')
     from tqdm import tqdm
     import torch
 
@@ -154,7 +153,6 @@ def evaluate_objective(self) -> ArrayLike:
             "max_steps": 100,
         }
     
-    # ks_init = torch.load('/net/beegfs-hpc/work/lagemannk/container/hydrogym_dev2/home/firedrake/hydrogym_rllib_multiEnv/hydrogym/hydrogym/torch_env/ks_init.tensor')
     env = OneDimEnv(config)
 
     action_space = env.action_space 
@@ -186,9 +184,5 @@ def evaluate_objective(self) -> ArrayLike:
     plt.savefig('/net/beegfs-hpc/work/lagemannk/container/workspace_christian_ext/logs/result.png', bbox_inches='tight')
     plt.close()
 
-    env.reset()
-
-
-
-    
+    env.reset()    
   
\ No newline at end of file