Move the UI interface into a new UIStage class which can be added to pipelines like a regular Stage