update example to be the same as in paper