Add ability to set the number of fit iterations