Removed the printing of the dataset after each iteration