Added number of iterations as hyper parameter