YH-269: Styling fixes