Updated the wizard example