improve the ui
improve the ui