Improved UI Controller API