diff --git a/cara/mc-output.py b/cara/mc-output.py
index 9651e468..1a60e9bb 100644
--- a/cara/mc-output.py
+++ b/cara/mc-output.py
@@ -1,7 +1,8 @@
 from cara.montecarlo import *
 from cara.model_scenarios import *
 
-# compare_concentration_curves([classroom_model, classroom_model_with_hepa], ['Just window', 'Window and HEPA'])
+classroom_model.infection_probability()
+compare_concentration_curves([classroom_model, classroom_model_with_hepa], ['Just window', 'Window and HEPA'])
 
 #print(np.mean(chorale_model.infection_probability()))
 #print(np.mean(chorale_model.infection_probability())+np.std(chorale_model.infection_probability()))
@@ -11,12 +12,12 @@ from cara.model_scenarios import *
 
 
 
-#print(np.mean(classroom_model_with_hepa.infection_probability()))
-composite_plot_pi_vs_viral_load([shared_office_worst_model[1], shared_office_model[1],shared_office_better_model[1]],
-                                labels=['No mask &\nwindows closed', 'Baseline', 'Baseline +\nHEPA filter'],
-                                colors=['tomato', '#1f77b4', 'limegreen'],
-                                title='$P(I|qID)$ vs $vl$ - Shared office scenario',
-                                vl_points=200)
+# #print(np.mean(classroom_model_with_hepa.infection_probability()))
+# composite_plot_pi_vs_viral_load([shared_office_worst_model[1], shared_office_model[1],shared_office_better_model[1]],
+#                                 labels=['No mask &\nwindows closed', 'Baseline', 'Baseline +\nHEPA filter'],
+#                                 colors=['tomato', '#1f77b4', 'limegreen'],
+#                                 title='$P(I|qID)$ vs $vl$ - Shared office scenario',
+#                                 vl_points=200)
 #plot_pi_vs_viral_load([shared_office_model[1]], labels=['Baseline, qID=60', 'HEPA, qID=60', 'No mask + windows closed, qID=60'],title='$P(I|qID)$ - Shared office scenario')