Adds unit creation to the wizard