Update tests for README.md sample code