Setup the general UI