Add imports to example