Add import to example code in README