improved the UI