some modification to the UI