From b1c5b1c51807dc6b771a0f3c7d709970a9c5401f Mon Sep 17 00:00:00 2001
From: Alex_Hubert <alex.hubert@etu.unistra.fr>
Date: Fri, 5 Nov 2021 21:51:45 +0100
Subject: [PATCH] HPC

---
 lib/objects.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/lib/objects.py b/lib/objects.py
index 893f3b8..3c92dee 100755
--- a/lib/objects.py
+++ b/lib/objects.py
@@ -157,5 +157,8 @@ class System:
         self.update_jp()
         self.update(dt)
         self.time = self.time + dt
+        for body in self.bodylist:
+            body.p = body.v*body.m
+
+        return 1
 
-        return 1
\ No newline at end of file