Update example in docs