Hook up more of the UI