Updated code example.