Fixes to example