More styling to match the design