add option to change number of rounds