Updated code blocks