Duh, adding import keyword
Cleaning up examples