Update example set-up code