Improved code block styling