diff --git a/ch06/02_bonus_additional-experiments/additional-experiments.py b/ch06/02_bonus_additional-experiments/additional-experiments.py
index a56224e..b9e824d 100644
--- a/ch06/02_bonus_additional-experiments/additional-experiments.py
+++ b/ch06/02_bonus_additional-experiments/additional-experiments.py
@@ -254,7 +254,7 @@ if __name__ == "__main__":
         help=(
             "Whether to use 'pretrained' or 'random' weights."
         )
-    )    
+    )
     parser.add_argument(
         "--trainable_layers",
         type=str,
@@ -290,7 +290,6 @@ if __name__ == "__main__":
     else:
         raise ValueError("Invalid --trainable_token argument")
 
-
     ###############################
     # Load model
     ###############################