Updated wizard code.