change mnist iterator