diff --git a/src/1_einstein_model_solutions.md b/src/1_einstein_model_solutions.md
index 7e335ceb9844e717f829e624e68edff5dff5bda8..4f7f13c1ee2e19fc9fc2ef02adbd73192c05f32e 100644
--- a/src/1_einstein_model_solutions.md
+++ b/src/1_einstein_model_solutions.md
@@ -30,6 +30,7 @@ configure_plotting()
 1. An ideal gas only contains 3 positional degrees of freedom.
 2. $C = 2k_B$.
 3. See image below
+
 ```python
 fig, ax = plt.subplots()
 omega = np.linspace(0.1, 3)
@@ -46,8 +47,10 @@ ax.set_yticks([0,1, 2])
 ax.set_yticklabels(['$0$','$1$', '$2$'])
 ax.set_title('$T_1<T_2$')
 ax.legend()
-draw_classic_axes(ax, xlabeloffset=.2);
+draw_classic_axes(ax, xlabeloffset=.2)
+fig.show();
 ```
+
 4. Minus sign in the exponent. This would result in $n_B(T = 0) = -1$, which is not physical.
 5. See plot with slider