updated to follow comments in blog post