move UI initialisation into the UI class