Improved styling of code blocks