Improves the ui a bit