Added imports to example