Updated code block styling