Updated pywizard script