Updated code block to fix syntax