Added iteration count checking
Changed UI