Update code example with correct output