Updated code block