diff --git a/agents/saved_models/A0 b/agents/saved_models/A0
index 994a351..86a47df 100644
Binary files a/agents/saved_models/A0 and b/agents/saved_models/A0 differ
diff --git a/agents/saved_models/C0 b/agents/saved_models/C0
index 87c2d30..23a1fbf 100644
Binary files a/agents/saved_models/C0 and b/agents/saved_models/C0 differ
diff --git a/entities/player.py b/entities/player.py
index a3694dc..b8bd4c1 100644
--- a/entities/player.py
+++ b/entities/player.py
@@ -170,7 +170,6 @@ class Player(pygame.sprite.Sprite):
 
             1 - 1/(np.exp((self.stats.health -
                    self.stats.stats['health'])/self.stats.stats['health']))
-            if not self.is_dead() > 0 else -1
         ]
 
         self.state_features = [
diff --git a/figures/actor_loss.png b/figures/actor_loss.png
index 031ad58..13822a0 100644
Binary files a/figures/actor_loss.png and b/figures/actor_loss.png differ
diff --git a/figures/critic_loss.png b/figures/critic_loss.png
index c6cac6b..b6642a2 100644
Binary files a/figures/critic_loss.png and b/figures/critic_loss.png differ
diff --git a/figures/score.png b/figures/score.png
index d5d1bb6..43e863a 100644
Binary files a/figures/score.png and b/figures/score.png differ
diff --git a/figures/total_loss.png b/figures/total_loss.png
index 2ff68c3..da4dcf6 100644
Binary files a/figures/total_loss.png and b/figures/total_loss.png differ