Improve main.c template for wizard.