Fix: correct data loading in test.plot_iteration()

Also clarified text messages and plot.