Add support for code blocks