UI adjustments including adding color