diff --git a/single-agent.py b/single-agent.py
index 956a00b..37f0161 100644
--- a/single-agent.py
+++ b/single-agent.py
@@ -61,16 +61,22 @@ for i in tqdm(range(n_episodes)):
     if np.mean(avg_score) > np.mean(best_score):
         print(
             f"\nNew Best score: {np.mean(avg_score)}\
-            \nOld Best score: {np.mean(best_score)}")
+            \nOld Best score: {np.mean(best_score)}"
+        )
         best_score = avg_score
         print("Saving models for agent...")
-        game.level.player_sprites[0].agent.save_models(
+        agent.save_models(
             actr_chkpt="player_actor", crtc_chkpt="player_critic")
         print("Models saved ...\n")
+    else:
+        print(
+            f"Average score of round: {np.mean(avg_score)}\
+              \nBest score: {np.mean(best_score)}"
+        )
 
 
 print("\nEpisodes done, saving models...")
-game.level.player_sprites[0].agent.save_models(
+agent.save_models(
     actr_chkpt="player_actor", crtc_chkpt="player_critic")
 print("Models saved ...\n")
 
diff --git a/tmp/ppo/player_actor b/tmp/ppo/player_actor
new file mode 100644
index 0000000..c046887
Binary files /dev/null and b/tmp/ppo/player_actor differ
diff --git a/tmp/ppo/player_critic b/tmp/ppo/player_critic
new file mode 100644
index 0000000..cbfc2f7
Binary files /dev/null and b/tmp/ppo/player_critic differ