diff --git a/BLcourse2.3/01_one_dim.py b/BLcourse2.3/01_one_dim.py
index 2b766ee27082e9f6c998d20729fad0d28e0d08d7..e9e55f233fc306eb2da38ca98ce710d5a359daa0 100644
--- a/BLcourse2.3/01_one_dim.py
+++ b/BLcourse2.3/01_one_dim.py
@@ -410,7 +410,9 @@ with torch.no_grad():
         if ii == 1:
             ax.legend()
     ax_sigmas.legend()
+# -
 
+# +
 # When running as script
 if not is_interactive():
     plt.show()
diff --git a/BLcourse2.3/02_two_dim.py b/BLcourse2.3/02_two_dim.py
index cde9cc7b0e6d4eb5b1c6ca479b272afc9c2fdf88..b7ee2273a279ef69e86f6d47fb1ac71d4fb13f97 100644
--- a/BLcourse2.3/02_two_dim.py
+++ b/BLcourse2.3/02_two_dim.py
@@ -370,6 +370,8 @@ ax.set_zlim((contour_z, zlim[1] + abs(contour_z)))
 ax.contourf(data_pred.XG, data_pred.YG, y_std, zdir="z", offset=contour_z)
 # -
 
+# +
 # When running as script
 if not is_interactive():
     plt.show()
+# -