Tweaked the ui a bit