Added maximum number of training epochs