add imports to example to make it compile