Update README.md and added code