Integrating UI with mediator pattern (in-progress)