Added the UI for the controller