Updated code example