Fixing some issues with the blog post and code