Improved the styling