update example to match paper