Updated the README.md and added the 2 classes required