diff --git a/Master_Thesis_IKEA_final_handin.pdf b/Thesis.pdf
similarity index 100%
rename from Master_Thesis_IKEA_final_handin.pdf
rename to Thesis.pdf
diff --git a/recommenders/ikea/training/trainBidirGRU.py b/recommenders/ikea/training/trainBidirGRU.py
index 4568483..cf4d1d4 100644
--- a/recommenders/ikea/training/trainBidirGRU.py
+++ b/recommenders/ikea/training/trainBidirGRU.py
@@ -59,8 +59,7 @@ def train_BidirGRU4Rec(
     output_tokenizer=None,
 ):
     """
-    Trains GRU4Rec model with provided arguments.
-    Also used to pretrain embedding layer for diversity rewards
+    Trains Bidirectional-GRU4Rec model with provided arguments.
     """
 
     disable_pro_bar = not progress_bar
diff --git a/recommenders/ikea/training/trainGRU.py b/recommenders/ikea/training/trainGRU.py
index 428bb71..ebf0c43 100644
--- a/recommenders/ikea/training/trainGRU.py
+++ b/recommenders/ikea/training/trainGRU.py
@@ -59,7 +59,6 @@ def train_GRU4Rec(
 ):
     """
     Trains GRU4Rec model with provided arguments.
-    Also used to pretrain embedding layer for diversity rewards
     """
 
     disable_pro_bar = not progress_bar
diff --git a/recommenders/ikea/training/trainSMORL.py b/recommenders/ikea/training/trainSMORL.py
index df77d4c..92bd827 100644
--- a/recommenders/ikea/training/trainSMORL.py
+++ b/recommenders/ikea/training/trainSMORL.py
@@ -62,7 +62,7 @@ def train_SMORL(
     save_pretrained_embeddings=True,
 ):
     """
-    Trains smorl model with provided arguments.
+    Trains SMORL model with provided arguments.
     """
 
     disable_pro_bar = not progress_bar