tweaked ui a bit