Fix UI colors and styles