improve the UI