Added iterative training