Move ui code block to theme file.