Fixes to the example