Revise code to reflect revised blog post