Add parameters in init function