Add custom code block rendering support