Added plot_iteration_summary to __init__