diff --git a/Logs_alg1717262897.pkl b/Logs_alg1717262897.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..be113231d7f3a504ce73f493717016a5571b64f7
Binary files /dev/null and b/Logs_alg1717262897.pkl differ
diff --git a/Logs_env1717262884.pkl b/Logs_env1717262884.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..2ffa797c0facc6a957ebf69bebd5167a3e86bb61
Binary files /dev/null and b/Logs_env1717262884.pkl differ
diff --git a/Logs_env1717262892.pkl b/Logs_env1717262892.pkl
new file mode 100644
index 0000000000000000000000000000000000000000..2ffa797c0facc6a957ebf69bebd5167a3e86bb61
Binary files /dev/null and b/Logs_env1717262892.pkl differ
diff --git a/ionen_env/__pycache__/__init__.cpython-311.pyc b/ionen_env/__pycache__/__init__.cpython-311.pyc
index c28dd447b716b3dc3ca1a4dd89c5e5c79f158ec0..6c05e9b44f82b7361b73958ffad5dcecab3baf23 100644
Binary files a/ionen_env/__pycache__/__init__.cpython-311.pyc and b/ionen_env/__pycache__/__init__.cpython-311.pyc differ
diff --git a/ionen_env/envs/__pycache__/Ionenbeschl.cpython-311.pyc b/ionen_env/envs/__pycache__/Ionenbeschl.cpython-311.pyc
index 70da98ada6274a8b619b7f3b15119b45a15fec9e..601824366645bbf5c2afff0e4337742883a23aa5 100644
Binary files a/ionen_env/envs/__pycache__/Ionenbeschl.cpython-311.pyc and b/ionen_env/envs/__pycache__/Ionenbeschl.cpython-311.pyc differ
diff --git a/ionen_env/envs/__pycache__/__init__.cpython-311.pyc b/ionen_env/envs/__pycache__/__init__.cpython-311.pyc
index 8ba7f2d88be810cfa9c4e89c1b8bc0d86a5df78d..97eeabc04b3ff97a363510f6aacb6a0a8d4d6918 100644
Binary files a/ionen_env/envs/__pycache__/__init__.cpython-311.pyc and b/ionen_env/envs/__pycache__/__init__.cpython-311.pyc differ
diff --git a/main.py b/main.py
index 0e3077058c857e9b83b7bca8d1a276c60d2b440a..6e907c87ea0bd695b55fc881b29ee29422093743 100644
--- a/main.py
+++ b/main.py
@@ -62,7 +62,7 @@ def torchrlAlg1():
 
     frames_per_batch = 1000
     # For a complete training, bring the number of frames up to 1M
-    total_frames = 10_000
+    total_frames = 1_000_000
 
     sub_batch_size = 64  # cardinality of the sub-samples gathered from the current data in the inner loop
     num_epochs = 10  # optimization steps per batch of data collected
@@ -282,10 +282,10 @@ def torchrlAlg1():
         # this is a nice-to-have but nothing necessary for PPO to work.
         scheduler.step()
 
-    # torch.save({'policy_state_dict': policy_module.state_dict(),
-    #             'value_state_dict': value_module.state_dict(),
-    #             'loss_state_dict': loss_module.state_dict()},
-    #            MODELPATH)
+    torch.save({'policy_state_dict': policy_module.state_dict(),
+                'value_state_dict': value_module.state_dict(),
+                'loss_state_dict': loss_module.state_dict()},
+               MODELPATH)
 
     # These modules are not good to load
     #             'collector_state_dict': collector.state_dict(),
diff --git a/model1.pth b/model1.pth
index e6c89ce1cb516e76b7d680db6c04021aad0d914e..9a8e52e876a1622e81ab41d9c9c73e9f0d3e5091 100644
Binary files a/model1.pth and b/model1.pth differ