Cleans up the code to match the blog post