Change prototype of main