add option to specify team size