added plotTreatmentEffects function