updated README.md and added code