Better code block styling