Get UI to work with model