diff --git a/DuelingDQN/Dueling_DQN_Discrete.py b/DuelingDQN/DuelingDQN_Discrete.py
similarity index 97%
rename from DuelingDQN/Dueling_DQN_Discrete.py
rename to DuelingDQN/DuelingDQN_Discrete.py
index 9ac4a12..4cef6b5 100644
--- a/DuelingDQN/Dueling_DQN_Discrete.py
+++ b/DuelingDQN/DuelingDQN_Discrete.py
@@ -10,7 +10,7 @@
 import random
 
 tf.keras.backend.set_floatx('float64')
-# wandb.init(name='DuelingDQN', project="deep-rl-tf2")
+wandb.init(name='DuelingDQN', project="deep-rl-tf2")
 
 parser = argparse.ArgumentParser()
 parser.add_argument('--gamma', type=float, default=0.95)
@@ -117,7 +117,7 @@ def train(self, max_episodes=1000):
                 state = next_state
             self.target_update()
             print('EP{} EpisodeReward={}'.format(ep, total_reward))
-            # wandb.log({'Reward': total_reward})
+            wandb.log({'Reward': total_reward})
 
 
 def main():