Update README.md

Changed the class name and added module import.