improved the ui a bit