Add mediation models and output