Fix parameters in main