Updated code section