Add import to example

Closes #224