Add the UI feature