Added missing import to the example in README. Fixes #7