Update setup_wizard.py