Change styling to match the mockup