Updated code chunk