Update README.md

For training