Updating code snippets in README.