Update with latest code sample from repo.