[PDI-11302] - Styling tweaks based on UX feedback