Clean up the styling