Update styling for code blocks