diff --git a/solution.py b/solution.py
index a5f0c6f..1111482 100644
--- a/solution.py
+++ b/solution.py
@@ -1055,12 +1055,13 @@ def visualize_color_attribution_and_counterfactual(
# %% [markdown]
#
Checkpoint 4
# At this point you have:
-# - Created a StarGAN that can change the class of an image
-# - Evaluated the StarGAN on unseen data
-# - Used the StarGAN to create counterfactual images
-# - Used the counterfactual images to highlight the differences between classes
-# - Interpolated between the images to see how the classifier behaves
-#
+#
+# - Created a StarGAN that can change the class of an image
+# - Evaluated the StarGAN on unseen data
+# - Used the StarGAN to create counterfactual images
+# - Used the counterfactual images to highlight the differences between classes
+# - Interpolated between the images to see how the classifier behaves
+#
# %% [markdown]
# # Part 5: Exploring the Style Space, finding the answer
# So color is important... but not always? What's going on!?
@@ -1166,11 +1167,13 @@ def visualize_color_attribution_and_counterfactual(
#
Checkpoint 5
# Congratulations! You have made it to the end of the exercise!
# You have:
-# - Created a StarGAN that can change the class of an image
-# - Evaluated the StarGAN on unseen data
-# - Used the StarGAN to create counterfactual images
-# - Used the counterfactual images to highlight the differences between classes
-# - Used the style space to understand the differences between classes
+#
+# - Created a StarGAN that can change the class of an image
+# - Evaluated the StarGAN on unseen data
+# - Used the StarGAN to create counterfactual images
+# - Used the counterfactual images to highlight the differences between classes
+# - Used the style space to understand the differences between classes
+#
#
# If you have any questions, feel free to ask them in the chat!
# And check the Solutions exercise for a definite answer to how these classes are defined!