Update README.md

Add missing import statement