Change .code structure
Add .code component