diff --git a/dqn_learning.py b/dqn_learning.py
index fe5597a..762305e 100644
--- a/dqn_learning.py
+++ b/dqn_learning.py
@@ -296,7 +296,7 @@ def main():
 
     save_model(best_score_model, model, results)
 
-    print("Whole process took: {}", time.time() - start_time_all)
+    print("Whole process took: {} seconds".format(time.time() - start_time_all))
 
 def save_model(best_score_model, last_model, parameters_dict):
     path = "results_" + datetime.now().strftime("%Y%m%d_%H_%M")
diff --git a/test.py b/test.py
index 735b983..10b2091 100644
--- a/test.py
+++ b/test.py
@@ -46,7 +46,7 @@ def main():
         print("Episode: {}, rewards: {}, time: {}".format(episode, reward_sum, time.time() - start_time), end="\n")
         rewards.append(reward_sum)
 
-    print("reward: {}".format(sum(rewards)/len(rewards)))
+    print("Average reward: {}".format(sum(rewards)/len(rewards)))
 
 
 if __name__ == '__main__':