diff --git a/validation.py b/validation.py
index 1d84c67a93216c8d19b5c4b7df2cfca401fcb855..de8dd96e7784cf8cf730721bed2b7af9c459ef53 100644
--- a/validation.py
+++ b/validation.py
@@ -18,15 +18,15 @@ class Validation:
         total_turns = 0
 
         for _ in range(num_games):
-            if strategy == 'optimal':
+            if strategy == 'Optimal':
                 turns = self.play_optimal_strategy()
-            elif strategy == 'dice1':
+            elif strategy == 'SafeDice':
                 turns = self.play_dice_strategy(1)
-            elif strategy == 'dice2':
+            elif strategy == 'NormalDice':
                 turns = self.play_dice_strategy(2)
-            elif strategy == 'dice3':
+            elif strategy == 'RiskyDice':
                 turns = self.play_dice_strategy(3)
-            elif strategy == 'random':
+            elif strategy == 'Random':
                 turns = self.play_random_strategy()
 
             total_turns += turns
@@ -69,18 +69,18 @@ class Validation:
         return turns
 
     def compare_strategies(self, num_games=1000):
-        optimal_cost = self.simulate_game(strategy='optimal', num_games=num_games)
-        dice1_cost = self.simulate_game(strategy='dice1', num_games=num_games)
-        dice2_cost = self.simulate_game(strategy='dice2', num_games=num_games)
-        dice3_cost = self.simulate_game(strategy='dice3', num_games=num_games)
-        random_cost = self.simulate_game(strategy='random', num_games=num_games)
+        optimal_cost = self.simulate_game(strategy='Optimal', num_games=num_games)
+        dice1_cost = self.simulate_game(strategy='SafeDice', num_games=num_games)
+        dice2_cost = self.simulate_game(strategy='NormalDice', num_games=num_games)
+        dice3_cost = self.simulate_game(strategy='RiskyDice', num_games=num_games)
+        random_cost = self.simulate_game(strategy='Random', num_games=num_games)
 
         return {
-            'optimal': optimal_cost,
-            'dice1': dice1_cost,
-            'dice2': dice2_cost,
-            'dice3': dice3_cost,
-            'random': random_cost
+            'Optimal': optimal_cost,
+            'SafeDice': dice1_cost,
+            'NormalDice': dice2_cost,
+            'RiskyDice': dice3_cost,
+            'Random': random_cost
         }
 
     def play_one_turn(self, dice_choice, cur_pos, prison):