Adding code in MediatorImpl