Improve code block formatting