Fix the styling