Added missing import to example code in README.