Update example imports

[mcastorina]

I noticed these two imports were missing when I tried to run the example.

Btw, thanks for the nice library!