[Yujun] Further simplified the example.