Ready for blog

Updated the code ready for the blog post