Add code in README.md