This contains the UI