CSS changes to fix display bug