improved code example in README.md