Update to code for latest blog post.