Update the example.
Add the model code