Move wizard code out of dialog_main.py