enhance the ui