Add styling fix