Small modification
Changed number of iterations