Update code snippit in README.md