Add option to print iteration