make the code match the blog post

make the code match the blog post