Use correct import paths

This is now a copy-pastable example.
Merge pull request #3 from beatgammit/patch-1

Use correct import paths