update code sample with import