Missed import on README example usage code.
Merge pull request #49 from yoloseem/patch-1

Missed import on README example usage code.