Minor UI adjustments