Update example
Add some tests