Modified code example