Updated the code blocks