CSS changes to fix UI