Updated example as well

Update example