add import to example code