Clean up code block rendering