Updating Mediator Pattern class