Updated example in docs