Update styling for code block