Minor changes to code blocks