add training iteration to imporve performance