Update README.md

Changed the import statement to make it more flexible