Code example fix

Fixed missing import.
Merge pull request #10 from johnraz/patch-1

Code example fix