Add CodeBlock class