Improved code example in README.md